35 lines
910 B
Python
35 lines
910 B
Python
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)
|