debug prints

weni-ai · Dec 27, 2024 · 8b57004 · 8b57004
1 parent cb086e5
commit 8b57004
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/nexus/zeroshot/client.py b/nexus/zeroshot/client.py
@@ -110,13 +110,14 @@ def _invoke_runpod(self, prompt):
         return response
 
     def _invoke_zeroshot(self, model_backend: str):
+        print("[ + INVOKE ZEROSHOT + ]")
         return {
             "runpod": self._invoke_runpod,
             "bedrock": self._invoke_bedrock
         }.get(model_backend)
 
     def _invoke_function_calling(self):
-
+        print("[ + INVOKE FUNCTION CALLING + ]")
         classifier = ChatGPTFunctionClassifier(
             agent_goal=self.zeroshot_data.get("context"),
         )
@@ -140,6 +141,7 @@ def _invoke_function_calling(self):
         formatted_classification = classification_formater.get_classification(self.zeroshot_data)
 
         response = {"output": formatted_classification}
+        print("Function calling response: ", response)
         return response
 
     def invoke(self):

diff --git a/router/clients/zeroshot.py b/router/clients/zeroshot.py
@@ -12,7 +12,7 @@ def __init__(self, prompt: str) -> None:
         self.prompt = prompt
 
     def fast_predict(self, message: str, actions: List[Dict], language: str = "por"):
-        print("[+ Calling Zeroshot in Nexus +]")
+        print("[+ NexusZeroshotClient FastPredict +]")
         zeroshot_data = {
             'context': self.prompt,
             'language': language,