Update mcp/openai_utils.py

mcp/openai_utils.py

@@ -6,10 +6,10 @@ import os
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 async def ai_summarize(text: str, prompt: str = None):
-    """Summarize and synthesize results using OpenAI GPT-4o."""
     if not prompt:
         prompt = "Summarize these biomedical search results, highlighting key findings and future research directions:"
-    response = openai.ChatCompletion.create(
+    client = openai.AsyncOpenAI(api_key=openai.api_key)
+    response = await client.chat.completions.create(
         model="gpt-4o",
         messages=[
             {"role": "system", "content": "You are an expert biomedical research assistant."},
@@ -17,11 +17,11 @@ async def ai_summarize(text: str, prompt: str = None):
         ],
         max_tokens=350
     )
-    return response['choices'][0]['message']['content']
+    return response.choices[0].message.content
 
 async def ai_qa(question: str, context: str = ""):
-    """Answer a user question using OpenAI GPT-4o, with provided context."""
-    response = openai.ChatCompletion.create(
+    client = openai.AsyncOpenAI(api_key=openai.api_key)
+    response = await client.chat.completions.create(
         model="gpt-4o",
         messages=[
             {"role": "system", "content": "You are an advanced biomedical research agent."},
@@ -29,4 +29,4 @@ async def ai_qa(question: str, context: str = ""):
         ],
         max_tokens=350
     )
-    return response['choices'][0]['message']['content']
+    return response.choices[0].message.content