|
|
from openai import OpenAI |
|
|
import os |
|
|
|
|
|
class BasicAgent: |
|
|
def __init__(self): |
|
|
api_key = os.getenv("OPENAI_API_KEY") |
|
|
if not api_key: |
|
|
raise ValueError("OPENAI_API_KEY environment variable not set.") |
|
|
self.client = OpenAI(api_key=api_key) |
|
|
print("BasicAgent (GPT-based) initialized.") |
|
|
|
|
|
def __call__(self, question: str) -> str: |
|
|
print(f"Agent received question (first 50 chars): {question[:50]}...") |
|
|
try: |
|
|
response = self.client.chat.completions.create( |
|
|
model="gpt-4", |
|
|
messages=[ |
|
|
{"role": "system", "content": "You are a helpful assistant."}, |
|
|
{"role": "user", "content": question} |
|
|
], |
|
|
temperature=0.7, |
|
|
max_tokens=500, |
|
|
) |
|
|
answer = response.choices[0].message.content.strip() |
|
|
print(f"Agent returning answer: {answer[:100]}...") |
|
|
return answer |
|
|
except Exception as e: |
|
|
print(f"Error during OpenAI completion: {e}") |
|
|
return f"[ERROR from agent: {str(e)}]" |
|
|
|
