traceai-agno 0.1.0__tar.gz

traceai_agno-0.1.0/.gitignore

@@ -0,0 +1,85 @@
+*.py[cod]
+
+# C extensions
+*.so
+
+# IDEs
+.idea/
+.vscode/
+
+# Packages
+*.egg
+*.egg-info
+.eggs/
+# build
+parts
+bin
+var
+sdist
+dist
+develop-eggs
+.installed.cfg
+lib
+lib64
+
+# Installer logs
+pip-log.txt
+
+# Unit test / coverage reports
+.coverage
+.tox
+nosetests.xml
+
+# Complexity
+output/*.html
+output/*/index.html
+
+# Sphinx
+docs/_build
+
+# Cookiecutter
+output/
+
+.workon
+
+**.wav
+**.joblib
+
+**/__pycache__
+
+#Excel files
+*.xlsx
+#csv files
+*.csv
+#virtual environments
+**/.venv
+**/env
+**/.env
+*.DS_Store
+*.pdf
+**/node_modules
+**/dist
+*.log
+.DS_Store
+**/package-lock.json
+**/tsconfig.tsbuildinfo
+**/.fi
+**/.pytest_cache
+planning/
+
+# Java build artifacts
+**/target/
+**/.settings/
+**/.project
+**/.classpath
+**/.factorypath
+*.class
+*.jar
+*.war
+
+# C# build artifacts
+**/bin/
+**/obj/
+
+# Python
+**/uv.lock

traceai_agno-0.1.0/PKG-INFO

@@ -0,0 +1,263 @@
+Metadata-Version: 2.4
+Name: traceai-agno
+Version: 0.1.0
+Summary: TraceAI instrumentation for Agno Agent Framework
+Project-URL: Homepage, https://github.com/futureagi/traceai
+Project-URL: Documentation, https://docs.futureagi.com/traceai
+Project-URL: Repository, https://github.com/futureagi/traceai
+Author-email: FutureAGI <support@futureagi.com>
+License-Expression: Apache-2.0
+Keywords: agents,agno,ai,llm,observability,openinference,opentelemetry,traceai,tracing
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: System :: Monitoring
+Requires-Python: >=3.10
+Requires-Dist: fi-instrumentation-otel>=0.1.11
+Requires-Dist: opentelemetry-api>=1.0.0
+Requires-Dist: opentelemetry-exporter-otlp>=1.0.0
+Requires-Dist: opentelemetry-sdk>=1.0.0
+Provides-Extra: agno
+Requires-Dist: agno>=2.0.0; extra == 'agno'
+Requires-Dist: openinference-instrumentation-agno>=0.1.0; extra == 'agno'
+Provides-Extra: dev
+Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# TraceAI Agno Instrumentation
+
+OpenTelemetry instrumentation for [Agno](https://github.com/agno-agi/agno), the high-performance AI agent framework.
+
+## Installation
+
+```bash
+pip install traceai-agno
+```
+
+For full functionality, also install the Agno framework and its OpenInference instrumentation:
+
+```bash
+pip install agno openinference-instrumentation-agno
+```
+
+## Quick Start
+
+```python
+from fi_instrumentation import register
+from fi_instrumentation.fi_types import ProjectType
+from traceai_agno import configure_agno_tracing
+
+# Setup TraceAI
+trace_provider = register(
+    project_type=ProjectType.OBSERVE,
+    project_name="agno-agent",
+)
+
+# Configure Agno to use TraceAI (call BEFORE creating agents)
+configure_agno_tracing(tracer_provider=trace_provider)
+
+# Now import and use Agno normally
+from agno.agent import Agent
+from agno.models.openai import OpenAIChat
+
+agent = Agent(
+    model=OpenAIChat(id="gpt-4"),
+    description="A helpful assistant",
+)
+
+response = agent.run("What is the capital of France?")
+print(response.content)
+```
+
+## Configuration Options
+
+### Using fi_instrumentation (Recommended)
+
+```python
+from fi_instrumentation import register
+from fi_instrumentation.fi_types import ProjectType
+from traceai_agno import configure_agno_tracing
+
+trace_provider = register(
+    project_type=ProjectType.OBSERVE,
+    project_name="my-agno-project",
+)
+
+configure_agno_tracing(tracer_provider=trace_provider)
+```
+
+### Direct OTLP Configuration
+
+```python
+from traceai_agno import configure_agno_tracing
+
+configure_agno_tracing(
+    otlp_endpoint="https://api.traceai.com/v1/traces",
+    otlp_headers={"Authorization": "Bearer YOUR_API_KEY"},
+    project_name="my-agno-project",
+)
+```
+
+### Using Custom Tracer Provider
+
+```python
+from traceai_agno import setup_traceai_exporter, configure_agno_tracing
+
+# Create a custom tracer provider
+provider = setup_traceai_exporter(
+    endpoint="https://api.traceai.com/v1/traces",
+    headers={"Authorization": "Bearer YOUR_API_KEY"},
+    service_name="my-agno-service",
+    use_batch_processor=True,
+)
+
+configure_agno_tracing(tracer_provider=provider)
+```
+
+## Features
+
+### Agent Tracing
+
+Automatically captures:
+- Agent name, type, and configuration
+- Model information (provider, ID, temperature, etc.)
+- Tool count and configuration
+- Memory and knowledge settings
+- Debug mode and markdown settings
+
+### Tool Tracing
+
+Tracks tool executions including:
+- Tool name and description
+- Input parameters
+- Execution results
+
+### Team Tracing
+
+For multi-agent setups:
+- Team name
+- Team member agents
+- Inter-agent communication
+
+### Workflow Tracing
+
+For complex workflows:
+- Workflow name
+- Step execution
+- State transitions
+
+## Span Attributes
+
+The instrumentation adds the following attributes to spans:
+
+### LLM Attributes (OpenTelemetry GenAI Semantic Conventions)
+- `gen_ai.system` - Model provider (openai, anthropic, etc.)
+- `gen_ai.request.model` - Requested model ID
+- `gen_ai.request.temperature` - Temperature setting
+- `gen_ai.request.max_tokens` - Max tokens setting
+- `gen_ai.usage.input_tokens` - Input token count
+- `gen_ai.usage.output_tokens` - Output token count
+
+### Agent Attributes
+- `agent.name` - Agent name
+- `agent.type` - Agent type
+- `agent.description` - Agent description
+- `agent.instructions` - Agent instructions
+
+### Agno-Specific Attributes
+- `agno.agent.id` - Agent ID
+- `agno.tool_count` - Number of tools
+- `agno.team.name` - Team name
+- `agno.team.members` - Team member names
+- `agno.workflow.name` - Workflow name
+- `agno.session.id` - Session ID
+- `agno.user.id` - User ID
+- `agno.debug_mode` - Debug mode status
+- `agno.memory.enabled` - Memory enabled status
+- `agno.knowledge.enabled` - Knowledge enabled status
+
+## Helper Functions
+
+### Extract Agent Attributes
+
+```python
+from traceai_agno import get_agent_attributes
+
+agent = Agent(model=OpenAIChat(id="gpt-4"), name="MyAgent")
+attrs = get_agent_attributes(agent)
+# {'agent.name': 'MyAgent', 'agent.type': 'Agent', ...}
+```
+
+### Extract Tool Attributes
+
+```python
+from traceai_agno import get_tool_attributes
+
+def my_tool(query: str) -> str:
+    """Search for information."""
+    return "result"
+
+attrs = get_tool_attributes(my_tool)
+# {'gen_ai.tool.name': 'my_tool', 'gen_ai.tool.description': 'Search for information.'}
+```
+
+### Extract Team Attributes
+
+```python
+from traceai_agno import get_team_attributes
+
+team = Team(name="ResearchTeam", agents=[agent1, agent2])
+attrs = get_team_attributes(team)
+# {'agno.team.name': 'ResearchTeam', 'agno.team.members': 'Agent1, Agent2'}
+```
+
+### Detect Model Provider
+
+```python
+from traceai_agno import get_model_provider
+
+provider = get_model_provider("gpt-4")  # Returns "openai"
+provider = get_model_provider("claude-3-sonnet")  # Returns "anthropic"
+provider = get_model_provider("ollama/llama3")  # Returns "ollama"
+```
+
+### Create Trace Context
+
+```python
+from traceai_agno import create_trace_context
+
+context = create_trace_context(
+    session_id="session-123",
+    user_id="user-456",
+    tags=["production"],
+    metadata={"environment": "prod"},
+)
+```
+
+## Examples
+
+See the [examples](./examples) directory for complete usage examples:
+
+- `basic_agent.py` - Simple agent with tracing
+- `agent_with_tools.py` - Agent with tool calling
+- `team_example.py` - Multi-agent team coordination
+- `workflow_example.py` - Complex workflow tracing
+
+## Requirements
+
+- Python >= 3.10
+- opentelemetry-api >= 1.0.0
+- opentelemetry-sdk >= 1.0.0
+- opentelemetry-exporter-otlp >= 1.0.0
+- fi-instrumentation >= 0.1.0
+
+## License
+
+Apache-2.0

traceai_agno-0.1.0/README.md

@@ -0,0 +1,230 @@
+# TraceAI Agno Instrumentation
+
+OpenTelemetry instrumentation for [Agno](https://github.com/agno-agi/agno), the high-performance AI agent framework.
+
+## Installation
+
+```bash
+pip install traceai-agno
+```
+
+For full functionality, also install the Agno framework and its OpenInference instrumentation:
+
+```bash
+pip install agno openinference-instrumentation-agno
+```
+
+## Quick Start
+
+```python
+from fi_instrumentation import register
+from fi_instrumentation.fi_types import ProjectType
+from traceai_agno import configure_agno_tracing
+
+# Setup TraceAI
+trace_provider = register(
+    project_type=ProjectType.OBSERVE,
+    project_name="agno-agent",
+)
+
+# Configure Agno to use TraceAI (call BEFORE creating agents)
+configure_agno_tracing(tracer_provider=trace_provider)
+
+# Now import and use Agno normally
+from agno.agent import Agent
+from agno.models.openai import OpenAIChat
+
+agent = Agent(
+    model=OpenAIChat(id="gpt-4"),
+    description="A helpful assistant",
+)
+
+response = agent.run("What is the capital of France?")
+print(response.content)
+```
+
+## Configuration Options
+
+### Using fi_instrumentation (Recommended)
+
+```python
+from fi_instrumentation import register
+from fi_instrumentation.fi_types import ProjectType
+from traceai_agno import configure_agno_tracing
+
+trace_provider = register(
+    project_type=ProjectType.OBSERVE,
+    project_name="my-agno-project",
+)
+
+configure_agno_tracing(tracer_provider=trace_provider)
+```
+
+### Direct OTLP Configuration
+
+```python
+from traceai_agno import configure_agno_tracing
+
+configure_agno_tracing(
+    otlp_endpoint="https://api.traceai.com/v1/traces",
+    otlp_headers={"Authorization": "Bearer YOUR_API_KEY"},
+    project_name="my-agno-project",
+)
+```
+
+### Using Custom Tracer Provider
+
+```python
+from traceai_agno import setup_traceai_exporter, configure_agno_tracing
+
+# Create a custom tracer provider
+provider = setup_traceai_exporter(
+    endpoint="https://api.traceai.com/v1/traces",
+    headers={"Authorization": "Bearer YOUR_API_KEY"},
+    service_name="my-agno-service",
+    use_batch_processor=True,
+)
+
+configure_agno_tracing(tracer_provider=provider)
+```
+
+## Features
+
+### Agent Tracing
+
+Automatically captures:
+- Agent name, type, and configuration
+- Model information (provider, ID, temperature, etc.)
+- Tool count and configuration
+- Memory and knowledge settings
+- Debug mode and markdown settings
+
+### Tool Tracing
+
+Tracks tool executions including:
+- Tool name and description
+- Input parameters
+- Execution results
+
+### Team Tracing
+
+For multi-agent setups:
+- Team name
+- Team member agents
+- Inter-agent communication
+
+### Workflow Tracing
+
+For complex workflows:
+- Workflow name
+- Step execution
+- State transitions
+
+## Span Attributes
+
+The instrumentation adds the following attributes to spans:
+
+### LLM Attributes (OpenTelemetry GenAI Semantic Conventions)
+- `gen_ai.system` - Model provider (openai, anthropic, etc.)
+- `gen_ai.request.model` - Requested model ID
+- `gen_ai.request.temperature` - Temperature setting
+- `gen_ai.request.max_tokens` - Max tokens setting
+- `gen_ai.usage.input_tokens` - Input token count
+- `gen_ai.usage.output_tokens` - Output token count
+
+### Agent Attributes
+- `agent.name` - Agent name
+- `agent.type` - Agent type
+- `agent.description` - Agent description
+- `agent.instructions` - Agent instructions
+
+### Agno-Specific Attributes
+- `agno.agent.id` - Agent ID
+- `agno.tool_count` - Number of tools
+- `agno.team.name` - Team name
+- `agno.team.members` - Team member names
+- `agno.workflow.name` - Workflow name
+- `agno.session.id` - Session ID
+- `agno.user.id` - User ID
+- `agno.debug_mode` - Debug mode status
+- `agno.memory.enabled` - Memory enabled status
+- `agno.knowledge.enabled` - Knowledge enabled status
+
+## Helper Functions
+
+### Extract Agent Attributes
+
+```python
+from traceai_agno import get_agent_attributes
+
+agent = Agent(model=OpenAIChat(id="gpt-4"), name="MyAgent")
+attrs = get_agent_attributes(agent)
+# {'agent.name': 'MyAgent', 'agent.type': 'Agent', ...}
+```
+
+### Extract Tool Attributes
+
+```python
+from traceai_agno import get_tool_attributes
+
+def my_tool(query: str) -> str:
+    """Search for information."""
+    return "result"
+
+attrs = get_tool_attributes(my_tool)
+# {'gen_ai.tool.name': 'my_tool', 'gen_ai.tool.description': 'Search for information.'}
+```
+
+### Extract Team Attributes
+
+```python
+from traceai_agno import get_team_attributes
+
+team = Team(name="ResearchTeam", agents=[agent1, agent2])
+attrs = get_team_attributes(team)
+# {'agno.team.name': 'ResearchTeam', 'agno.team.members': 'Agent1, Agent2'}
+```
+
+### Detect Model Provider
+
+```python
+from traceai_agno import get_model_provider
+
+provider = get_model_provider("gpt-4")  # Returns "openai"
+provider = get_model_provider("claude-3-sonnet")  # Returns "anthropic"
+provider = get_model_provider("ollama/llama3")  # Returns "ollama"
+```
+
+### Create Trace Context
+
+```python
+from traceai_agno import create_trace_context
+
+context = create_trace_context(
+    session_id="session-123",
+    user_id="user-456",
+    tags=["production"],
+    metadata={"environment": "prod"},
+)
+```
+
+## Examples
+
+See the [examples](./examples) directory for complete usage examples:
+
+- `basic_agent.py` - Simple agent with tracing
+- `agent_with_tools.py` - Agent with tool calling
+- `team_example.py` - Multi-agent team coordination
+- `workflow_example.py` - Complex workflow tracing
+
+## Requirements
+
+- Python >= 3.10
+- opentelemetry-api >= 1.0.0
+- opentelemetry-sdk >= 1.0.0
+- opentelemetry-exporter-otlp >= 1.0.0
+- fi-instrumentation >= 0.1.0
+
+## License
+
+Apache-2.0

traceai_agno-0.1.0/examples/agent_with_tools.py

@@ -0,0 +1,136 @@
+"""Agno agent with tools example with TraceAI instrumentation.
+
+This example demonstrates how to set up tracing for an agent with tool calling.
+"""
+
+import os
+from typing import Optional
+
+# Setup tracing BEFORE importing Agno
+from fi_instrumentation import register
+from fi_instrumentation.fi_types import ProjectType
+from traceai_agno import configure_agno_tracing
+
+# Initialize TraceAI
+trace_provider = register(
+    project_type=ProjectType.OBSERVE,
+    project_name="agno-tools-example",
+)
+
+# Configure Agno to use TraceAI
+configure_agno_tracing(tracer_provider=trace_provider)
+
+# Now import Agno modules
+from agno.agent import Agent
+from agno.models.openai import OpenAIChat
+from agno.tools import tool
+
+
+# Define tools
+@tool
+def get_weather(city: str) -> str:
+    """Get the current weather for a city.
+
+    Args:
+        city: The name of the city to get weather for.
+
+    Returns:
+        A string describing the current weather.
+    """
+    # Mock weather data
+    weather_data = {
+        "new york": "Sunny, 72°F (22°C)",
+        "london": "Cloudy, 59°F (15°C)",
+        "tokyo": "Rainy, 68°F (20°C)",
+        "paris": "Partly cloudy, 65°F (18°C)",
+        "sydney": "Clear, 77°F (25°C)",
+    }
+    city_lower = city.lower()
+    return weather_data.get(city_lower, f"Weather data not available for {city}")
+
+
+@tool
+def calculate(expression: str) -> str:
+    """Evaluate a mathematical expression.
+
+    Args:
+        expression: A mathematical expression to evaluate (e.g., "2 + 2", "10 * 5").
+
+    Returns:
+        The result of the calculation.
+    """
+    try:
+        # Only allow safe mathematical operations
+        allowed_chars = set("0123456789+-*/.(). ")
+        if not all(c in allowed_chars for c in expression):
+            return "Error: Invalid characters in expression"
+        result = eval(expression)
+        return str(result)
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+
+@tool
+def search_knowledge(query: str, topic: Optional[str] = None) -> str:
+    """Search for information about a topic.
+
+    Args:
+        query: The search query.
+        topic: Optional topic to narrow down the search.
+
+    Returns:
+        Relevant information about the query.
+    """
+    # Mock knowledge base
+    knowledge = {
+        "python": "Python is a high-level programming language known for readability.",
+        "machine learning": "Machine learning is a subset of AI focused on learning from data.",
+        "agno": "Agno is a high-performance AI agent framework for building intelligent agents.",
+    }
+
+    query_lower = query.lower()
+    for key, value in knowledge.items():
+        if key in query_lower:
+            return value
+    return f"No specific information found for: {query}"
+
+
+def main():
+    """Run an agent with tools example."""
+    # Create an agent with tools
+    agent = Agent(
+        name="ToolAgent",
+        model=OpenAIChat(id="gpt-4"),
+        description="An assistant that can check weather, perform calculations, and search for information.",
+        tools=[get_weather, calculate, search_knowledge],
+        instructions=[
+            "Use the available tools to help answer questions",
+            "For weather questions, use the get_weather tool",
+            "For math problems, use the calculate tool",
+            "For information queries, use the search_knowledge tool",
+        ],
+        show_tool_calls=True,
+        markdown=True,
+    )
+
+    # Example queries that use different tools
+    queries = [
+        "What's the weather like in Paris?",
+        "What is 25 * 4 + 10?",
+        "Tell me about machine learning",
+        "What's the weather in Tokyo and calculate 100 / 5",
+    ]
+
+    print("Agent: ToolAgent")
+    print("Tools: get_weather, calculate, search_knowledge")
+    print("=" * 50)
+
+    for query in queries:
+        print(f"\nQuery: {query}")
+        print("-" * 40)
+        response = agent.run(query)
+        print(f"Response: {response.content}")
+
+
+if __name__ == "__main__":
+    main()