Merge pull request #166 from Chainlit/willy/openai-agents

feat: add openai-agents tracing processor

Author: willydouhard

examples/attachment.py

@@ -32,7 +32,7 @@ async def main():
                 "url": "https://api.github.com/repos/chainlit/chainlit",
                 "mime": "application/json",
                 "metadata": {"test": "test"},
-            }
+            },
         )
 
         print(attachment.to_dict())

examples/langchain_toolcall.py

@@ -1,15 +1,12 @@
-from literalai import LiteralClient
-
-from langchain_openai import ChatOpenAI  # type: ignore
+from dotenv import load_dotenv
+from langchain.agents import AgentExecutor, create_tool_calling_agent
+from langchain.agents.agent import BaseSingleActionAgent
 from langchain_community.tools.tavily_search import TavilySearchResults
-
-from langchain.agents import create_tool_calling_agent
-from langchain.agents import AgentExecutor
 from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.runnables.config import RunnableConfig
-from langchain.agents.agent import BaseSingleActionAgent
+from langchain_openai import ChatOpenAI  # type: ignore
 
-from dotenv import load_dotenv
+from literalai import LiteralClient
 
 # Add OPENAI_API_KEY and TAVILY_API_KEY for this example.
 load_dotenv()

examples/langchain_variable.py

@@ -1,8 +1,7 @@
+from dotenv import load_dotenv
 from langchain.chat_models import init_chat_model
-from literalai import LiteralClient
-
 
-from dotenv import load_dotenv
+from literalai import LiteralClient
 
 load_dotenv()
 

examples/llamaindex.py

@@ -1,6 +1,7 @@
-from literalai import LiteralClient
-from llama_index.core import Document, VectorStoreIndex
 from dotenv import load_dotenv
+from llama_index.core import Document, VectorStoreIndex
+
+from literalai import LiteralClient
 
 load_dotenv()
 
@@ -14,7 +15,7 @@
 questions = [
     "Tell me about LLMs",
     "How do you fine-tune a neural network ?",
-    "What is RAG ?"
+    "What is RAG ?",
 ]
 
 # No context, create a Thread (it will be named after the first user query)

examples/llamaindex_workflow.py

@@ -1,12 +1,8 @@
 import asyncio
-from llama_index.core.workflow import (
-    Event,
-    StartEvent,
-    StopEvent,
-    Workflow,
-    step,
-)
+
+from llama_index.core.workflow import Event, StartEvent, StopEvent, Workflow, step
 from llama_index.llms.openai import OpenAI
+
 from literalai.client import LiteralClient
 
 lai_client = LiteralClient()
@@ -16,7 +12,8 @@
 class JokeEvent(Event):
     joke: str
 
-class RewriteJoke(Event): 
+
+class RewriteJoke(Event):
     joke: str
 
 

examples/main.py

@@ -24,7 +24,7 @@ def get_completion(welcome_message, text):
             {
                 "role": "system",
                 "content": "Tell an inspiring quote to the user, mentioning their name. Be extremely supportive while "
-                           "keeping it short. Write one sentence per line.",
+                "keeping it short. Write one sentence per line.",
             },
             {
                 "role": "assistant",

examples/multimodal.py

@@ -1,12 +1,11 @@
 import base64
-import requests  # type: ignore
 import time
 
-from literalai import LiteralClient
-from openai import OpenAI
-
+import requests  # type: ignore
 from dotenv import load_dotenv
+from openai import OpenAI
 
+from literalai import LiteralClient
 
 load_dotenv()
 

examples/openai_agents.py

@@ -0,0 +1,27 @@
+import asyncio
+
+from agents import Agent, Runner, set_trace_processors, trace
+from dotenv import load_dotenv
+
+from literalai import LiteralClient
+
+load_dotenv()
+
+client = LiteralClient()
+
+
+async def main():
+    agent = Agent(name="Joke generator", instructions="Tell funny jokes.")
+
+    with trace("Joke workflow"):
+        first_result = await Runner.run(agent, "Tell me a joke")
+        second_result = await Runner.run(
+            agent, f"Rate this joke: {first_result.final_output}"
+        )
+        print(f"Joke: {first_result.final_output}")
+        print(f"Rating: {second_result.final_output}")
+
+
+if __name__ == "__main__":
+    set_trace_processors([client.openai_agents_tracing_processor()])
+    asyncio.run(main())

literalai/api/__init__.py

@@ -1,4 +1,4 @@
-from literalai.api.synchronous import LiteralAPI
 from literalai.api.asynchronous import AsyncLiteralAPI
+from literalai.api.synchronous import LiteralAPI
 
 __all__ = ["LiteralAPI", "AsyncLiteralAPI"]

literalai/api/helpers/attachment_helpers.py

@@ -1,9 +1,8 @@
 import mimetypes
 from typing import Dict, Optional, TypedDict, Union
 
-from literalai.observability.step import Attachment
-
 from literalai.api.helpers import gql
+from literalai.observability.step import Attachment
 
 
 def create_attachment_helper(