Merge branch 'main' into feat/new-langchain-callback-handler

AgentOps-AI · Jan 24, 2025 · ff352cd · ff352cd
2 parents 002bcbd + 0883c34
commit ff352cd
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 279 deletions.
diff --git a/agentops/host_env.py b/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:

diff --git a/agentops/llms/providers/ollama.py b/agentops/llms/providers/ollama.py
@@ -26,7 +26,7 @@ def handle_stream_chunk(chunk: dict):
 
             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 @@ def generator():
             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"]

diff --git a/examples/anthropic_examples/anthropic-example-sync.py b/examples/anthropic_examples/anthropic-example-sync.py
diff --git a/examples/openai_examples/openai_example_sync.py b/examples/openai_examples/openai_example_sync.py