Fix MockChatModel bug

ls1intum · Oct 29, 2024 · e2e9097 · e2e9097
1 parent af5ae94
commit e2e9097
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/intelligence-service/app/model.py b/server/intelligence-service/app/model.py
@@ -7,11 +7,11 @@
 model: BaseChatModel
 
 if os.getenv("GITHUB_ACTIONS") == "true":
-    # mock model for testing environment
-    class MockChatModel(BaseChatModel):
+    class MockChatModel():
         def invoke(self, message: str):
             return "Mock response"
     model = MockChatModel()
+
 elif settings.is_openai_available:
     model = ChatOpenAI()
 elif settings.is_azure_openai_available: