This commit is contained in:
Yiran Wu
2023-12-26 13:29:01 -05:00
committed by GitHub
parent 5b420f1d9a
commit 9b7a42fc57

View File

@@ -11,6 +11,7 @@ from .. import OpenAIWrapper
from ..code_utils import DEFAULT_MODEL, UNKNOWN, content_str, execute_code, extract_code, infer_lang
from ..function_utils import get_function_schema, load_basemodels_if_needed, serialize_to_str
from .agent import Agent
from .._pydantic import model_dump
try:
from termcolor import colored
@@ -640,7 +641,7 @@ class ConversableAgent(Agent):
# TODO: line 301, line 271 is converting messages to dict. Can be removed after ChatCompletionMessage_to_dict is merged.
extracted_response = client.extract_text_or_completion_object(response)[0]
if not isinstance(extracted_response, str):
extracted_response = extracted_response.model_dump(mode="dict")
extracted_response = model_dump(extracted_response)
return True, extracted_response
async def a_generate_oai_reply(