Update openai_api.py

openai_api.py

@@ -21,14 +21,11 @@ def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
     return result.data[0].embedding
 
 
-def get_response(prompt: str, model: str=COMPLETIONS_MODEL) -> str:
+def get_response(messages: list[dict], model: str=COMPLETIONS_MODEL) -> str:
     """Chat completion using OpenAI's GPT models."""
     response = client.chat.completions.create(
         model=model,
-        messages=[
-            {"role": "system", "content": META_PROMPT},
-            {"role": "user", "content": prompt}
-        ],
+        messages=messages,
         temperature=0,
         max_completion_tokens=800
     )