AI butler Jarvis

agent_jarvis/jarvis/gpt/ai_function.py

@@ -0,0 +1,41 @@
+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)