feature(#30, #29, #59): resolved parameters passing issue on getCompl…

…etion Method.
ttt246 · May 28, 2023 · 3c1a286 · 3c1a286
1 parent d78a075
commit 3c1a286
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/router/api.py b/src/router/api.py
@@ -43,7 +43,7 @@ def send_notification():
         token = data["token"]
         uuid = data["uuid"]
 
-        result = getCompletion(query, uuid)
+        result = getCompletion(query=query, uuid=uuid)
 
         notification = {"title": "alert", "content": result}
 

diff --git a/src/service/llm/chat_service.py b/src/service/llm/chat_service.py
@@ -126,7 +126,9 @@ def chat_with_ai(
 
                 # TODO: use a model defined elsewhere, so that model can contain
                 # temperature and other settings we care about
-                return ChatResponseModel(handle_chat_completion(model=model, messages=current_context))
+                return ChatResponseModel(
+                    handle_chat_completion(model=model, messages=current_context)
+                )
             except Exception as e:
                 # TODO: When we switch to langchain, this is built in
                 logger.warn("Error: ", "API Rate Limit Reached. Waiting 10 seconds...")

diff --git a/tests/functional/test_recipes.py b/tests/functional/test_recipes.py
@@ -11,7 +11,7 @@
 
 def test_langchain():
     error = "Error happened while analyzing your prompt. Please ask me again"
-    result = getCompletion("open three.js website", TEST_UUID)
+    result = getCompletion(query="open three.js website", uuid=TEST_UUID)
     print(result)
     assert result != error