Adds AI functions.

These are no-code functions written by AI.
pull/17/head
Torantulino 2023-04-01 10:34:32 +01:00
parent a608d8dbdc
commit 8767018c1e
1 changed files with 16 additions and 0 deletions

16
scripts/ai_functions.py Normal file
View File

@ -0,0 +1,16 @@
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"]