42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
|
from typing import List
|
||
|
|
|
||
|
|
from jarvis import CFG
|
||
|
|
from jarvis.gpt import gpt
|
||
|
|
from jarvis.gpt.message import Message
|
||
|
|
from jarvis.logger import logger
|
||
|
|
|
||
|
|
|
||
|
|
async def acall_ai_function(function: str, args: list, description: str, model: str | None = None) -> str:
|
||
|
|
"""Call an AI function
|
||
|
|
|
||
|
|
This is a magic function that can do anything with no-code. See
|
||
|
|
https://github.com/Torantulino/AI-Functions for more info.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
function (str): The function to call
|
||
|
|
args (list): The arguments to pass to the function
|
||
|
|
description (str): The description of the function
|
||
|
|
model (str, optional): The model to use. Defaults to None.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
str: The response from the function
|
||
|
|
"""
|
||
|
|
if model is None:
|
||
|
|
model = CFG.small_llm_model
|
||
|
|
# For each arg, if any are None, convert to "None":
|
||
|
|
args = [str(arg) if arg is not None else "None" for arg in args]
|
||
|
|
# parse args to comma separated string
|
||
|
|
args: str = ", ".join(args)
|
||
|
|
messages: List[Message] = [
|
||
|
|
{
|
||
|
|
"role": "system",
|
||
|
|
"content": f"You are now the following python function: ```# {description}"
|
||
|
|
f"\n{function}```\n\nOnly respond with your `return` value.",
|
||
|
|
},
|
||
|
|
{"role": "user", "content": args},
|
||
|
|
]
|
||
|
|
|
||
|
|
logger.debug(str(messages))
|
||
|
|
|
||
|
|
return await gpt.acreate_chat_completion(model=model, messages=messages, temperature=0)
|