from openai import OpenAI

class LLMRequest(BaseModel):
user_query: str

class LLMResponse(BaseModel):
llm_response: str

@router.post("/llm-query")

def llm_query(body: LLMRequest) -> LLMResponse =
OPENAI_API_KEY = db.secrets.get("OPENAI_API_KEY")

client = OpenAI(api_key=OPENAI_API_KEY)

completion = client.chat.completions.create(
model="gpt-4-0125-preview",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": body.user_query}
])

llm_response = completion.choices[0].message.content

return LLMResponse(llm_response=llm_response)