mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-18 22:44:21 +01:00
650e2dcd6d
updated README.md to explain new config added Azure yaml loader to config class centralized model retrieval into config class this commit effectively combines and replaces #700 and #580
27 lines
827 B
Python
27 lines
827 B
Python
import openai
|
|
from config import Config
|
|
cfg = Config()
|
|
|
|
openai.api_key = cfg.openai_api_key
|
|
|
|
# Overly simple abstraction until we create something better
|
|
def create_chat_completion(messages, model=None, temperature=None, max_tokens=None)->str:
|
|
"""Create a chat completion using the OpenAI API"""
|
|
if cfg.use_azure:
|
|
response = openai.ChatCompletion.create(
|
|
deployment_id=cfg.get_azure_deployment_id_for_model(model),
|
|
model=model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens
|
|
)
|
|
else:
|
|
response = openai.ChatCompletion.create(
|
|
model=model,
|
|
messages=messages,
|
|
temperature=temperature,
|
|
max_tokens=max_tokens
|
|
)
|
|
|
|
return response.choices[0].message["content"]
|