[Fix] Partners integration tests

agentops/host_env.py

@@ -100,9 +100,9 @@ def get_ram_details():
     try:
         ram_info = psutil.virtual_memory()
         return {
-            "Total": f"{ram_info.total / (1024 ** 3):.2f} GB",
-            "Available": f"{ram_info.available / (1024 ** 3):.2f} GB",
-            "Used": f"{ram_info.used / (1024 ** 3):.2f} GB",
+            "Total": f"{ram_info.total / (1024**3):.2f} GB",
+            "Available": f"{ram_info.available / (1024**3):.2f} GB",
+            "Used": f"{ram_info.used / (1024**3):.2f} GB",
             "Percentage": f"{ram_info.percent}%",
         }
     except:

agentops/llms/providers/ollama.py

@@ -26,7 +26,7 @@
 
             if chunk.get("done"):
                 llm_event.end_timestamp = get_ISO_time()
-                llm_event.model = f'ollama/{chunk.get("model")}'
+                llm_event.model = f"ollama/{chunk.get('model')}"
                 llm_event.returns = chunk
                 llm_event.returns["message"] = llm_event.completion
                 llm_event.prompt = kwargs["messages"]
@@ -53,7 +53,7 @@
             return generator()
 
         llm_event.end_timestamp = get_ISO_time()
-        llm_event.model = f'ollama/{response["model"]}'
+        llm_event.model = f"ollama/{response['model']}"
         llm_event.returns = response
         llm_event.agent_id = check_call_stack_for_agent_id()
         llm_event.prompt = kwargs["messages"]

pyproject.toml

@@ -45,7 +45,7 @@ test = [
     "anthropic",
     "cohere",
     "litellm",
-    "ai21>=3.0.0",
+    "ai21>=2.0.0, <3.0.0",
     "groq",
     "ollama",
     "mistralai",
@@ -58,6 +58,7 @@ test = [
     # "crewai-tools @ git+https://github.com/crewAIInc/crewAI-tools.git@a14091abb24527c97ccfcc8539d529c8b4559a0f; python_version>='3.10'", 
     # ------------------------------------------------------------------------------------------------------------------------------------
     # ;;
+    "taskweaver @ git+https://github.com/microsoft/[email protected]; python_version>='3.10'",
     "autogen<0.4.0",
     "pytest-cov",
     "fastapi[standard]",