Files
Auto-GPT/scripts/ai_functions.py
Torantulino 8767018c1e Adds AI functions.
These are no-code functions written by AI.
2023-04-01 10:34:32 +01:00

17 lines
576 B
Python

from typing import List, Optional
import json
import openai
def call_ai_function(function, args, description, model = "gpt-4"):
# parse args to comma seperated string
args = ", ".join(args)
messages = [{"role": "system", "content": f"You are now the following python function: ```# {description}\n{function}```\n\nOnly respond with your `return` value."},{"role": "user", "content": args}]
response = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=0
)
return response.choices[0].message["content"]