| 4 | |
| 5 | |
| 6 | def openai_complete_if_cache( |
| 7 | model="gpt-4o-mini", prompt=None, system_prompt=None, history_messages=[], **kwargs |
| 8 | ) -> str: |
| 9 | openai_client = OpenAI() |
| 10 | |
| 11 | messages = [] |
| 12 | if system_prompt: |
| 13 | messages.append({"role": "system", "content": system_prompt}) |
| 14 | messages.extend(history_messages) |
| 15 | messages.append({"role": "user", "content": prompt}) |
| 16 | |
| 17 | response = openai_client.chat.completions.create( |
| 18 | model=model, messages=messages, **kwargs |
| 19 | ) |
| 20 | return response.choices[0].message.content |
| 21 | |
| 22 | |
| 23 | if __name__ == "__main__": |