hack for gpt-3.5

gpt.py

@@ -223,7 +223,7 @@ def __init__(self, assistant: Assistant, markdown: bool, show_price: bool):
  listeners.append(PriceChatListener(assistant))
 
  if os.environ.get("GPTCLI_ALLOW_CODE_EXECUTION") == "1":
- listeners.append(CodeInterpreterListener("python_eval"))
+ listeners.append(CodeInterpreterListener("python"))
 
  listener = CompositeChatListener(listeners)
  super().__init__(

gptcli/openai.py

@@ -7,7 +7,7 @@
 
 FUNCTIONS_SCHEMA = [
  {
- "name": "python_eval",
+ "name": "python",
  "description": "Evaluate an arbitrary Python snippet",
  "parameters": {
  "type": "object",

gptcli/session.py

@@ -127,8 +127,17 @@ def _handle_function_call(self, function_call: FunctionCall) -> Message:
  function_name = function_call.get("name", "null")
 
  function_result = None
+
  try:
- function_arguments = json.loads(function_call["arguments"])
+ arguments = function_call.get("arguments", "{}")
+ if arguments.startswith("{"):
+ function_arguments = json.loads(arguments)
+ else:
+ # HACK: gpt-3.5-turbo sometimes returns a string instead of a dict for python calls
+ function_arguments = {
+ "source": function_call.get("arguments", ""),
+ }
+
  function_result = self.listener.on_function_call(
  function_name, **function_arguments
  )

requirements.txt

@@ -7,7 +7,7 @@ prompt-toolkit==3.0.38
 pytest==7.3.1
 PyYAML==6.0
 rich==13.3.2
-tiktoken==0.3.3
+tiktoken==0.4.0
 tokenizers==0.13.3
 typing_extensions==4.5.0
 jupyter-client==8.2.0