diff --git a/src/e84_geoai_common/llm/models/converse.py b/src/e84_geoai_common/llm/models/converse.py index e741230..25c4395 100644 --- 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 index 97b1622..59f7d69 100644 --- 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 index 8f41bac..6242a92 100644 --- 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 index c619035..866d820 100644 --- 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="""