formatting

Element84 · Feb 20, 2025 · 2d07b2a · 2d07b2a
1 parent 7372636
commit 2d07b2a
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 9 deletions.
diff --git a/src/e84_geoai_common/llm/models/converse.py b/src/e84_geoai_common/llm/models/converse.py
@@ -33,11 +33,9 @@
     "Claude Instant": "anthropic.claude-instant-v1",
     "Claude 3.5 Haiku": "anthropic.claude-3-5-haiku-20241022-v1:0",
     "Claude 3.5 Sonnet v2": "anthropic.claude-3-5-sonnet-20241022-v2:0",
-    # "Nova Canvas": "amazon.nova-canvas-v1:0",
-    "Nova Lite": "us.amazon.nova-lite-v1:0", #added 'us' infront for invocation
+    "Nova Lite": "us.amazon.nova-lite-v1:0",
     "Nova Micro": "us.amazon.nova-micro-v1:0",
     "Nova Pro": "us.amazon.nova-pro-v1:0",
-    # "Nova Reel": "amazon.nova-reel-v1:0",
     "Llama 3.1 70B Instruct": "us.meta.llama3-1-70b-instruct-v1:0",
     "Llama 3.1 8B Instruct": "us.meta.llama3-1-8b-instruct-v1:0",
     "Llama 3.2 11B Vision Instruct": "us.meta.llama3-2-11b-instruct-v1:0",

diff --git a/tests/llm/models/test_claude.py b/tests/llm/models/test_claude.py
@@ -64,7 +64,6 @@ def encode_image_to_base64(image_path: str) -> str:
     return encoded_bytes.decode("utf-8")
 
 def test_image_input() -> None:
-    # Simulated response when processing an image
     expected_text_output = "cat"
     mock_response = claude_response_with_content(expected_text_output)
 
@@ -74,7 +73,6 @@ def test_image_input() -> None:
     image_path = str(Path(__file__).parent / "images/cat.webp")
     base64_string = encode_image_to_base64(image_path)
 
-    # Mock image content
     image_content = Base64ImageContent(media_type="image/webp", data=base64_string)
     prompt_text = TextContent(
         text="Report the animal in the picture and only that, in lowercase. I.e. dog"

diff --git a/tests/llm/models/test_converse.py b/tests/llm/models/test_converse.py
@@ -64,7 +64,6 @@ def encode_image_to_bytes(image_path: str) -> bytes:
     return base64_encoded
 
 def test_image_input() -> None:
-    # Simulated response when processing an image
     expected_text_output = "cat"
     mock_response = converse_response_with_content(expected_text_output)
 
@@ -74,7 +73,6 @@ def test_image_input() -> None:
     image_path = str(Path(__file__).parent / "images/cat.webp")
     base64_string = encode_image_to_bytes(image_path)
 
-    # Mock image content
     image_content = Base64ImageContent(media_type="image/webp", data=base64_string)
     prompt_text = TextContent(
         text = """

diff --git a/tests/llm/models/test_nova.py b/tests/llm/models/test_nova.py
@@ -50,7 +50,6 @@ def encode_image_to_base64(image_path: str) -> str:
     return encoded_bytes.decode("utf-8")
 
 def test_image_input() -> None:
-    # Simulated response when processing an image
     expected_text_output = "cat"
     mock_response = nova_response_with_content(expected_text_output)
 
@@ -60,7 +59,6 @@ def test_image_input() -> None:
     image_path = str(Path(__file__).parent / "images/cat.webp")
     base64_string = encode_image_to_base64(image_path)
 
-    # Mock image content
     image_content = Base64ImageContent(media_type="image/webp", data=base64_string)
     prompt_text = TextContent(
         text="""