| from openai import AzureOpenAI | |
| import os | |
| class GPTAgent: | |
| def __init__(self, model_name): | |
| self.client = AzureOpenAI( | |
| api_key=os.getenv('AZURE_OPENAI_KEY'), | |
| api_version=os.getenv('AZURE_OPENAI_VERSION'), | |
| azure_endpoint=os.getenv('AZURE_OPENAI_ENDPOINT') | |
| ) | |
| self.deployment_name = model_name | |
| def invoke(self, text, **kwargs): | |
| response = self.client.chat.completions.create( | |
| model=self.deployment_name, | |
| messages=[ | |
| {"role": "system", "content": "You are a helpful assistant."}, | |
| {"role": "user", "content": text} | |
| ], | |
| **kwargs | |
| ) | |
| return response.choices[0].message.content | |