OpenInference Integration

from judgeval.tracer import Tracer
from openinference.instrumentation.openai import OpenAIInstrumentor
from openai import OpenAI

tracer = Tracer(project_name="multi_workflow_app")
OpenAIInstrumentor().instrument()

client = OpenAI()

def analyze_sentiment(text: str) -> str:
    response = client.chat.completions.create(
        model="gpt-5-mini",
        messages=[
            {"role": "system", "content": "You are a sentiment analysis expert."},
            {"role": "user", "content": f"Analyze the sentiment of: {text}"}
        ]
    )
    return response.choices[0].message.content

def extract_entities(text: str) -> str:
    response = client.chat.completions.create(
        model="gpt-5-mini",
        messages=[
            {"role": "system", "content": "You are an entity extraction expert."},
            {"role": "user", "content": f"Extract entities from: {text}"}
        ]
    )
    return response.choices[0].message.content

def summarize_text(text: str) -> str:
    response = client.chat.completions.create(
        model="gpt-5-mini",
        messages=[
            {"role": "system", "content": "You are a summarization expert."},
            {"role": "user", "content": f"Summarize: {text}"}
        ]
    )
    return response.choices[0].message.content

@tracer.observe(span_type="function")  
def main():
    text = "Apple Inc. announced today that their new iPhone will be released next month. The market reacted positively to this news."

    sentiment = analyze_sentiment(text)
    entities = extract_entities(text)
    summary = summarize_text(text)

    print(f"Sentiment: {sentiment}")
    print(f"Entities: {entities}")
    print(f"Summary: {summary}")

if __name__ == "__main__":
    main()