Fixing some dict value checking for function_call (#249)

letta-ai · Nov 6, 2023 · 8ad1209 · 8ad1209
1 parent 8aa2e53
commit 8ad1209
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/memgpt/local_llm/llm_chat_completion_wrappers/airoboros.py b/memgpt/local_llm/llm_chat_completion_wrappers/airoboros.py
@@ -133,7 +133,7 @@ def create_function_call(function_call):
             elif message["role"] == "assistant":
                 prompt += f"\nASSISTANT: {message['content']}"
                 # need to add the function call if there was one
-                if message["function_call"]:
+                if "function_call" in message and message["function_call"]:
                     prompt += f"\n{create_function_call(message['function_call'])}"
             elif message["role"] == "function":
                 # TODO find a good way to add this
@@ -339,7 +339,7 @@ def create_function_call(function_call, inner_thoughts=None):
                 prompt += f"\nASSISTANT:"
                 # need to add the function call if there was one
                 inner_thoughts = message["content"]
-                if message["function_call"]:
+                if "function_call" in message and message["function_call"]:
                     prompt += f"\n{create_function_call(message['function_call'], inner_thoughts=inner_thoughts)}"
             elif message["role"] == "function":
                 # TODO find a good way to add this

diff --git a/memgpt/local_llm/llm_chat_completion_wrappers/dolphin.py b/memgpt/local_llm/llm_chat_completion_wrappers/dolphin.py
@@ -165,7 +165,7 @@ def create_function_call(function_call):
                     prompt += f"\n{message['content']}"
                 # prompt += f"\nASSISTANT: {message['content']}"
                 # need to add the function call if there was one
-                if message["function_call"]:
+                if "function_call" in message and message["function_call"]:
                     prompt += f"\n{create_function_call(message['function_call'])}"
                 prompt += f"{IM_END_TOKEN}"
             elif message["role"] == "function":

diff --git a/memgpt/local_llm/llm_chat_completion_wrappers/zephyr.py b/memgpt/local_llm/llm_chat_completion_wrappers/zephyr.py
@@ -96,7 +96,7 @@ def create_function_call(function_call):
                     prompt += f"\n{message['content']}"
                 # prompt += f"\nASSISTANT: {message['content']}"
                 # need to add the function call if there was one
-                if message["function_call"]:
+                if "function_call" in message and message["function_call"]:
                     prompt += f"\n{create_function_call(message['function_call'])}"
                 prompt += f"{IM_END_TOKEN}"
             elif message["role"] == "function":
@@ -259,7 +259,7 @@ def create_function_call(function_call, inner_thoughts=None):
                 prompt += f"\n<|assistant|>"
                 # need to add the function call if there was one
                 inner_thoughts = message["content"]
-                if message["function_call"]:
+                if "function_call" in message and message["function_call"]:
                     prompt += f"\n{create_function_call(message['function_call'], inner_thoughts=inner_thoughts)}"
             elif message["role"] == "function":
                 # TODO find a good way to add this