Files
Auto-GPT/scripts/llm_utils.py
Taylor Brown ae9448cb89 Consolidate calls to openai
Starting to abstract away the calls to openai
2023-04-02 21:51:07 -05:00

17 lines
457 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:
response = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=temperature,
max_tokens=max_tokens
)
return response.choices[0].message["content"]