Added missing requirement, fixed test

Getty · Dec 28, 2024 · e510a38 · e510a38
1 parent 9802949
commit e510a38
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,6 +8,7 @@ authors = [
 dependencies = [
     "pydantic>=2.0.0",
     "typing-extensions>=4.0.0",
+    "pyyaml>=6.0.0"
 ]
 requires-python = ">=3.8"
 readme = "README.md"

diff --git a/tests/test_mcp.py b/tests/test_mcp.py
@@ -23,7 +23,7 @@ def test_001_openai(self):
             engine_model = engine.model('gpt-4o')
             self.assertIsInstance(engine_model, MCPModel)
             self.assertEqual(type(engine_model).__name__, "MCPModel")
-            engine_slash_model = engine.model('openai/gpt-4o')
+            engine_slash_model = MCP.model('openai/gpt-4o')
             self.assertIsInstance(engine_slash_model, MCPModel)
             self.assertEqual(type(engine_slash_model).__name__, "MCPModel")
             model = MCP.model('gpt-4o')
@@ -43,7 +43,7 @@ def test_002_anthropic(self):
             engine_model = engine.model('claude-2.1')
             self.assertIsInstance(engine_model, MCPModel)
             self.assertEqual(type(engine_model).__name__, "MCPModel")
-            engine_slash_model = engine.model('anthropic/claude-2.1')
+            engine_slash_model = MCP.model('anthropic/claude-2.1')
             self.assertIsInstance(engine_slash_model, MCPModel)
             self.assertEqual(type(engine_slash_model).__name__, "MCPModel")
             model = MCP.model('claude-2.1')
@@ -63,7 +63,7 @@ def test_003_groq(self):
             engine_model = engine.model('llama3-8b-8192')
             self.assertIsInstance(engine_model, MCPModel)
             self.assertEqual(type(engine_model).__name__, "MCPModel")
-            engine_slash_model = engine.model('groq/llama3-8b-8192')
+            engine_slash_model = MCP.model('groq/llama3-8b-8192')
             self.assertIsInstance(engine_slash_model, MCPModel)
             self.assertEqual(type(engine_slash_model).__name__, "MCPModel")
             model = MCP.model('llama3-8b-8192')