feat: add new examples from pocketflow-academy
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
import os
|
||||
from openai import OpenAI
|
||||
from pathlib import Path
|
||||
|
||||
# Get the project root directory (parent of utils directory)
|
||||
ROOT_DIR = Path(__file__).parent.parent
|
||||
|
||||
# Initialize OpenAI client with API key from environment
|
||||
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
|
||||
|
||||
def call_llm(prompt: str) -> str:
|
||||
"""Call OpenAI API to analyze text
|
||||
|
||||
Args:
|
||||
prompt (str): Input prompt for the model
|
||||
|
||||
Returns:
|
||||
str: Model response
|
||||
"""
|
||||
try:
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-4o",
|
||||
messages=[{"role": "user", "content": prompt}]
|
||||
)
|
||||
return response.choices[0].message.content
|
||||
|
||||
except Exception as e:
|
||||
print(f"Error calling LLM API: {str(e)}")
|
||||
return ""
|
||||
|
||||
if __name__ == "__main__":
|
||||
# Test LLM call
|
||||
response = call_llm("What is web search?")
|
||||
print("Response:", response)
|
||||
Reference in New Issue
Block a user