OMGeeky/gpt-pilot
1
0
Fork 0
mirror of https://github.com/OMGeeky/gpt-pilot.git synced 2026-01-06 19:29:33 +01:00
Code Issues Packages Projects Releases Wiki Activity

update code to support python <3.10, update requirements, fix readme issues

Browse Source
This commit is contained in:
LeonOstrez
2023-09-26 17:13:15 +01:00
parent c9afca290e
commit 459f9380a7
3 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pilot/utils/function_calling.py
View File

@@ -1,8 +1,9 @@
import json
import re
from typing import Literal, Optional, TypedDict, Callable
from typing import Union, TypeVar, List, Dict, Literal, Optional, TypedDict, Callable
JsonType = str | int | float | bool | None | list["JsonType"] | dict[str, "JsonType"]
JsonTypeBase = Union[str, int, float, bool, None, List["JsonType"], Dict[str, "JsonType"]]
JsonType = TypeVar("JsonType", bound=JsonTypeBase)
class FunctionParameters(TypedDict):
@@ -33,7 +34,7 @@ class FunctionCallSet(TypedDict):
functions: dict[str, Callable]
def add_function_calls_to_request(gpt_data, function_calls: FunctionCallSet | None):
def add_function_calls_to_request(gpt_data, function_calls: Union[FunctionCallSet, None]):
if function_calls is None:
return
@@ -57,7 +58,7 @@ def add_function_calls_to_request(gpt_data, function_calls: FunctionCallSet | No
})
def parse_agent_response(response, function_calls: FunctionCallSet | None):
def parse_agent_response(response, function_calls: Union[FunctionCallSet, None]):
"""
Post-processes the response from the agent.
@@ -177,7 +178,7 @@ class JsonPrompter:
self,
prompt: str,
functions: list[FunctionType],
function_to_call: str | None = None,
function_to_call: Union[str, None] = None,
) -> str:
"""Generate the llama prompt