fix: linting and README

Author: RyanDsilva

.github/workflows/publish.yml

@@ -16,7 +16,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v5
       with:
-        python-version: "3.14"
+        python-version: "3.13"
     - name: Install pypa/build
       run: >-
         python3 -m

.github/workflows/test.yml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
 
     steps:
     - uses: actions/checkout@v3

README.md

@@ -2,7 +2,7 @@
 
 [![PyPI version](https://badge.fury.io/py/strands-nova.svg)](https://badge.fury.io/py/strands-nova)
 [![Python Support](https://img.shields.io/pypi/pyversions/strands-nova.svg)](https://pypi.org/project/strands-nova/)
-[![Tests](https://github.com/aidendef/strands-nova/actions/workflows/test.yml/badge.svg)](https://github.com/aidendef/strands-nova/actions/workflows/test.yml)
+[![Tests](https://github.com/RyanDsilva/strands-nova/actions/workflows/test.yml/badge.svg)](https://github.com/RyanDsilva/strands-nova/actions/workflows/test.yml)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 
 Amazon Nova API model provider for [Strands Agents SDK](https://github.com/strands-agents/sdk-python)

examples/basic_usage.py

@@ -3,55 +3,48 @@
 
 import asyncio
 import os
+
 from dotenv import load_dotenv
+from strands import Agent
 
 from strands_nova import NovaModel
-from strands import Agent
 
 
 async def main():
     """Run basic example."""
     # Load environment variables
     load_dotenv()
-    
+
     # Check for API key
     if not os.getenv("NOVA_API_KEY"):
         print("Error: NOVA_API_KEY not found in environment variables.")
         print("Please set: export NOVA_API_KEY='your-api-key'")
         print("Get your API key from: https://nova.amazon.com/apis")
         return
-    
+
     print("=== Basic Nova API Example ===\n")
-    
+
     # Initialize model
-    model = NovaModel(
-        model="nova-premier-v1",
-        temperature=0.7,
-        max_tokens=1024
-    )
-    
+    model = NovaModel(model="nova-premier-v1", temperature=0.7, max_tokens=1024)
+
     # Create agent
     agent = Agent(model=model)
-    
+
     # Simple conversation
     print("Question: What are the key benefits of Amazon's Nova models?")
     response = await agent.invoke_async("What are the key benefits of Amazon's Nova models?")
     print(f"Response: {response.message}\n")
-    
+
     # Technical question
     print("Question: Explain how transformer architecture works in simple terms.")
     response = await agent.invoke_async("Explain how transformer architecture works in simple terms.")
     print(f"Response: {response.message}\n")
-    
+
     # Example with Nova Pro model
     print("\n=== Using Nova Pro v3 ===\n")
-    model_pro = NovaModel(
-        model="Nova Pro v3 (6.x)",
-        temperature=0.5,
-        max_tokens=512
-    )
+    model_pro = NovaModel(model="Nova Pro v3 (6.x)", temperature=0.5, max_tokens=512)
     agent_pro = Agent(model=model_pro)
-    
+
     print("Question: Write a Python function to calculate fibonacci numbers.")
     response = await agent_pro.invoke_async("Write a Python function to calculate fibonacci numbers.")
     print(f"Response: {response.message}\n")

examples/streaming_example.py

@@ -3,6 +3,7 @@
 
 import asyncio
 import os
+
 from dotenv import load_dotenv
 
 from strands_nova import NovaModel
@@ -11,48 +12,38 @@
 async def basic_streaming():
     """Demonstrate basic streaming."""
     print("=== Basic Streaming Example ===\n")
-    
-    model = NovaModel(
-        model="nova-premier-v1",
-        temperature=0.7,
-        max_tokens=500
-    )
-    
+
+    model = NovaModel(model="nova-premier-v1", temperature=0.7, max_tokens=500)
+
     print("Question: Write a short story about a robot learning to paint.\n")
     print("Response: ", end="", flush=True)
-    
+
     async for event in model.stream("Write a short story about a robot learning to paint."):
         if "contentBlockDelta" in event:
             delta = event["contentBlockDelta"]["delta"]
             if "text" in delta:
                 print(delta["text"], end="", flush=True)
-    
+
     print("\n")
 
 
 async def streaming_with_system_prompt():
     """Demonstrate streaming with system prompt."""
     print("\n=== Streaming with System Prompt ===\n")
-    
-    model = NovaModel(
-        model="nova-premier-v1",
-        temperature=0.3
-    )
-    
+
+    model = NovaModel(model="nova-premier-v1", temperature=0.3)
+
     system_prompt = "You are a helpful coding assistant. Provide concise, well-commented code examples."
-    
+
     print("Question: Write a Python function to merge two sorted lists.\n")
     print("Response: ", end="", flush=True)
-    
-    async for event in model.stream(
-        "Write a Python function to merge two sorted lists.",
-        system_prompt=system_prompt
-    ):
+
+    async for event in model.stream("Write a Python function to merge two sorted lists.", system_prompt=system_prompt):
         if "contentBlockDelta" in event:
             delta = event["contentBlockDelta"]["delta"]
             if "text" in delta:
                 print(delta["text"], end="", flush=True)
-    
+
     print("\n")
 
 
@@ -70,15 +61,13 @@ async def reasoning_model_example():
     model = NovaModel(
         model="nova-premier-v1",  # Placeholder until reasoning model is available
         # reasoning_effort="medium",  # Currently not supported, will raise error
-        temperature=0.5
+        temperature=0.5,
     )
 
     print("Question: What is the sum of all prime numbers between 1 and 100?\n")
     print("Response: ", end="", flush=True)
 
-    async for event in model.stream(
-        "What is the sum of all prime numbers between 1 and 100?"
-    ):
+    async for event in model.stream("What is the sum of all prime numbers between 1 and 100?"):
         if "contentBlockDelta" in event:
             delta = event["contentBlockDelta"]["delta"]
             if "text" in delta:
@@ -90,50 +79,46 @@ async def reasoning_model_example():
 async def web_search_example():
     """Demonstrate web search capabilities."""
     print("\n=== Web Search Example ===\n")
-    
-    model = NovaModel(
-        model="nova-premier-v1",
-        web_search_options={"search_context_size": "low"}
-    )
-    
+
+    model = NovaModel(model="nova-premier-v1", web_search_options={"search_context_size": "low"})
+
     print("Question: What is the current Amazon stock price?\n")
     print("Response: ", end="", flush=True)
-    
+
     async for event in model.stream(
-        "What is the current Amazon stock price?",
-        system_prompt="You are a helpful financial assistant."
+        "What is the current Amazon stock price?", system_prompt="You are a helpful financial assistant."
     ):
         if "contentBlockDelta" in event:
             delta = event["contentBlockDelta"]["delta"]
             if "text" in delta:
                 print(delta["text"], end="", flush=True)
-    
+
     print("\n")
 
 
 async def main():
     """Run all streaming examples."""
     # Load environment variables
     load_dotenv()
-    
+
     # Check for API key
     if not os.getenv("NOVA_API_KEY"):
         print("Error: NOVA_API_KEY not found in environment variables.")
         print("Please set: export NOVA_API_KEY='your-api-key'")
         print("Get your API key from: https://nova.amazon.com/apis")
         return
-    
+
     # Run examples
     await basic_streaming()
     await streaming_with_system_prompt()
-    
+
     # Optional: Run reasoning and web search examples
     # Note: These require specific model access
     try:
         await reasoning_model_example()
     except Exception as e:
         print(f"\nReasoning model example skipped: {e}\n")
-    
+
     try:
         await web_search_example()
     except Exception as e:

pyproject.toml

@@ -22,7 +22,6 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
-    "Programming Language :: Python :: 3.14",
     "Topic :: Software Development :: Libraries :: Python Modules",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
 ]
@@ -62,7 +61,7 @@ asyncio_default_fixture_loop_scope = "function"
 
 [tool.black]
 line-length = 120
-target-version = ["py310", "py311", "py312", "py314"]
+target-version = ["py310", "py311", "py312", "py313"]
 
 [tool.ruff]
 line-length = 120
@@ -72,7 +71,7 @@ select = ["E", "F", "I", "N", "W"]
 ignore = ["E501"]
 
 [tool.mypy]
-python_version = "3.14"
+python_version = "3.13"
 warn_return_any = true
 warn_unused_configs = true
 ignore_missing_imports = true