add phi data agent example

Author: cfahlgren1

examples/agents/phidata.py

@@ -0,0 +1,19 @@
+from phi.agent import Agent
+from phi.model.openai import OpenAIChat
+from phi.tools.duckduckgo import DuckDuckGo
+from observers.observers import wrap_openai
+from observers.stores import DatasetsStore
+from openai import OpenAI
+
+
+client = wrap_openai(OpenAI(), store=DatasetsStore(repo_name="websearch-agent-traces"))
+
+web_agent = Agent(
+    name="Web Agent",
+    model=OpenAIChat(id="gpt-4o", client=client),
+    tools=[DuckDuckGo()],
+    instructions=["Always include sources"],
+    show_tool_calls=True,
+    markdown=True,
+)
+web_agent.print_response("Whats happening in France?")

src/observers/base.py

@@ -2,15 +2,34 @@
 from abc import ABC, abstractmethod
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, Dict, List, Literal, Optional
+from typing_extensions import Literal
 
 if TYPE_CHECKING:
     from argilla import Argilla
 
 
+@dataclass
+class Function:
+    """Function tool call information"""
+
+    name: str
+    arguments: str
+
+
+@dataclass
+class ToolCall:
+    """Tool call information"""
+
+    id: str
+    type: Literal["function"]
+    function: Function
+
+
 @dataclass
 class Message:
     role: Literal["system", "user", "assistant", "function"]
     content: str
+    tool_calls: Optional[List[ToolCall]] = None
 
 
 @dataclass

src/observers/stores/migrations/001_create_schema_version.sql

@@ -9,7 +9,7 @@ CREATE TABLE IF NOT EXISTS openai_records (
     id VARCHAR PRIMARY KEY,
     model VARCHAR,
     timestamp TIMESTAMP,
-    messages STRUCT(role VARCHAR, content VARCHAR)[],
+    messages JSON,
     assistant_message TEXT,
     completion_tokens INTEGER,
     prompt_tokens INTEGER,