chore: general_function_schema, general_tool_choice -> GENERAL_FUNCTION_SCHEMA, GENERAL_TOOL_CHOICE

This commit is contained in:
刘棒棒 2023-11-21 17:14:39 +08:00
parent bec4bc0e83
commit 7a5c66b01e
2 changed files with 5 additions and 5 deletions

View file

@ -1,6 +1,6 @@
# function in tools, https://platform.openai.com/docs/api-reference/chat/create#chat-create-tools
# Reference: https://github.com/KillianLucas/open-interpreter/blob/v0.1.14/interpreter/llm/setup_openai_coding_llm.py
general_function_schema = {
GENERAL_FUNCTION_SCHEMA = {
"name": "execute",
"description": "Executes code on the user's machine, **in the users local environment**, and returns the output",
"parameters": {
@ -27,4 +27,4 @@ general_function_schema = {
# tool_choice value for general_function_schema
# https://platform.openai.com/docs/api-reference/chat/create#chat-create-tool_choice
general_tool_choice = {"type": "function", "function": {"name": "execute"}}
GENERAL_TOOL_CHOICE = {"type": "function", "function": {"name": "execute"}}

View file

@ -21,7 +21,7 @@ from tenacity import (
from metagpt.config import CONFIG
from metagpt.logs import logger
from metagpt.provider.base_gpt_api import BaseGPTAPI
from metagpt.provider.constant import general_function_schema, general_tool_choice
from metagpt.provider.constant import GENERAL_FUNCTION_SCHEMA, GENERAL_TOOL_CHOICE
from metagpt.schema import Message
from metagpt.utils.singleton import Singleton
from metagpt.utils.token_counter import (
@ -248,8 +248,8 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter):
"""
if "tools" not in kwargs:
configs = {
"tools": [{"type": "function", "function": general_function_schema}],
"tool_choice": general_tool_choice,
"tools": [{"type": "function", "function": GENERAL_FUNCTION_SCHEMA}],
"tool_choice": GENERAL_TOOL_CHOICE,
}
kwargs.update(configs)