From 2a4633a4b9762494c108630f61c92f306c61e8cb Mon Sep 17 00:00:00 2001 From: Pratyush Shukla Date: Mon, 2 Jun 2025 23:15:34 +0530 Subject: [PATCH 1/2] format --- agentops/instrumentation/google_adk/patch.py | 4 ++-- agentops/sdk/decorators/__init__.py | 1 + examples/openai_agents_sdk/customer_service_agent.ipynb | 2 +- examples/openai_agents_sdk/tool_usage_agent.ipynb | 2 +- examples/openai_agents_sdk/web_search_agent.ipynb | 2 +- 5 files changed, 6 insertions(+), 5 deletions(-) diff --git a/agentops/instrumentation/google_adk/patch.py b/agentops/instrumentation/google_adk/patch.py index a51a81c7b..5d3dff73e 100644 --- a/agentops/instrumentation/google_adk/patch.py +++ b/agentops/instrumentation/google_adk/patch.py @@ -278,14 +278,14 @@ def _extract_llm_attributes(llm_request_dict: dict, llm_response: Any) -> dict: if "prompt_tokens_details" in usage: for detail in usage["prompt_tokens_details"]: if "modality" in detail and "token_count" in detail: - attributes[f'gen_ai.usage.prompt_tokens.{detail["modality"].lower()}'] = detail[ + attributes[f"gen_ai.usage.prompt_tokens.{detail['modality'].lower()}"] = detail[ "token_count" ] if "candidates_tokens_details" in usage: for detail in usage["candidates_tokens_details"]: if "modality" in detail and "token_count" in detail: - attributes[f'gen_ai.usage.completion_tokens.{detail["modality"].lower()}'] = detail[ + attributes[f"gen_ai.usage.completion_tokens.{detail['modality'].lower()}"] = detail[ "token_count" ] diff --git a/agentops/sdk/decorators/__init__.py b/agentops/sdk/decorators/__init__.py index f775b45d5..608b23908 100644 --- a/agentops/sdk/decorators/__init__.py +++ b/agentops/sdk/decorators/__init__.py @@ -20,6 +20,7 @@ tool = create_entity_decorator(SpanKind.TOOL) operation = task + # For backward compatibility: @session decorator calls @trace decorator @functools.wraps(trace) def session(*args, **kwargs): diff --git a/examples/openai_agents_sdk/customer_service_agent.ipynb b/examples/openai_agents_sdk/customer_service_agent.ipynb index 95d61015f..e5507f6b6 100644 --- a/examples/openai_agents_sdk/customer_service_agent.ipynb +++ b/examples/openai_agents_sdk/customer_service_agent.ipynb @@ -7,7 +7,7 @@ "outputs": [], "source": [ "# IMPORTANT: After running this cell, restart the kernel before continuing.\n", - "get_ipython().system('pip install -q -U agentops openai-agents')" + "get_ipython().system(\"pip install -q -U agentops openai-agents\")" ] }, { diff --git a/examples/openai_agents_sdk/tool_usage_agent.ipynb b/examples/openai_agents_sdk/tool_usage_agent.ipynb index 659995ce9..bfd5b6e2b 100644 --- a/examples/openai_agents_sdk/tool_usage_agent.ipynb +++ b/examples/openai_agents_sdk/tool_usage_agent.ipynb @@ -7,7 +7,7 @@ "outputs": [], "source": [ "# IMPORTANT: After running this cell, restart the kernel before continuing.\n", - "get_ipython().system('pip install -q -U agentops openai-agents')" + "get_ipython().system(\"pip install -q -U agentops openai-agents\")" ] }, { diff --git a/examples/openai_agents_sdk/web_search_agent.ipynb b/examples/openai_agents_sdk/web_search_agent.ipynb index b7c2e18ed..35bf9616c 100644 --- a/examples/openai_agents_sdk/web_search_agent.ipynb +++ b/examples/openai_agents_sdk/web_search_agent.ipynb @@ -7,7 +7,7 @@ "outputs": [], "source": [ "# IMPORTANT: After running this cell, restart the kernel before continuing.\n", - "get_ipython().system('pip install -q -U agentops openai-agents')" + "get_ipython().system(\"pip install -q -U agentops openai-agents\")" ] }, { From 41a7bbc26221f5db377855081ce1bb605eadffd0 Mon Sep 17 00:00:00 2001 From: Pratyush Shukla Date: Mon, 2 Jun 2025 23:19:12 +0530 Subject: [PATCH 2/2] ruff --- agentops/sdk/decorators/__init__.py | 2 +- .../openai_agents_sdk/customer_service_agent.ipynb | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/agentops/sdk/decorators/__init__.py b/agentops/sdk/decorators/__init__.py index 608b23908..043679fa8 100644 --- a/agentops/sdk/decorators/__init__.py +++ b/agentops/sdk/decorators/__init__.py @@ -23,7 +23,7 @@ # For backward compatibility: @session decorator calls @trace decorator @functools.wraps(trace) -def session(*args, **kwargs): +def session(*args, **kwargs): # noqa: F811 """@deprecated Use @agentops.trace instead. Wraps the @trace decorator for backward compatibility.""" logger.info(colored("@agentops.session decorator is deprecated. Please use @agentops.trace instead.", "yellow")) # If called as @session or @session(...) diff --git a/examples/openai_agents_sdk/customer_service_agent.ipynb b/examples/openai_agents_sdk/customer_service_agent.ipynb index e5507f6b6..ed313b71c 100644 --- a/examples/openai_agents_sdk/customer_service_agent.ipynb +++ b/examples/openai_agents_sdk/customer_service_agent.ipynb @@ -34,7 +34,7 @@ "metadata": {}, "outputs": [], "source": [ - "from __future__ import annotations as _annotations\n", + "from __future__ import annotations as _annotations # noqa: F404\n", "\n", "import random\n", "import uuid\n", @@ -44,7 +44,7 @@ "# API key is automatically read from the environment variable\n", "agentops.init(tags=[\"customer-service-agent\", \"agentops-example\"])\n", "\n", - "from agents import (\n", + "from agents import ( # noqa: E402\n", " Agent,\n", " HandoffOutputItem,\n", " ItemHelpers,\n", @@ -58,7 +58,7 @@ " handoff,\n", " trace,\n", ")\n", - "from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX\n", + "from agents.extensions.handoff_prompt import RECOMMENDED_PROMPT_PREFIX # noqa: E402\n", "\n", "### CONTEXT\n", "\n", @@ -217,7 +217,7 @@ ], "metadata": { "kernelspec": { - "display_name": "base", + "display_name": "agentops (3.11.11)", "language": "python", "name": "python3" }, @@ -231,7 +231,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.12.7" + "version": "3.11.11" } }, "nbformat": 4,