removing logs

This commit is contained in:
João Moura
2024-09-23 20:56:58 -03:00
parent 9eeebd3c65
commit 44f74fd6f6
2 changed files with 0 additions and 9 deletions

View File

@@ -302,7 +302,6 @@ class ToolUsage:
) -> Union[ToolCalling, InstructorToolCalling]:
try:
if self.function_calling_llm:
print("self.function_calling_llm")
model = (
InstructorToolCalling
if self.function_calling_llm.supports_function_calling()
@@ -323,9 +322,7 @@ class ToolUsage:
),
max_attempts=1,
)
print("converter", converter)
tool_object = converter.to_pydantic()
print("tool_object", tool_object)
calling = ToolCalling(
tool_name=tool_object["tool_name"],
arguments=tool_object["arguments"],

View File

@@ -132,7 +132,6 @@ def handle_partial_json(
converter_cls: Optional[Type[Converter]] = None,
) -> Union[dict, BaseModel, str]:
match = re.search(r"({.*})", result, re.DOTALL)
print("handle_partial_json")
if match:
try:
exported_result = model.model_validate_json(match.group(0))
@@ -161,11 +160,8 @@ def convert_with_instructions(
agent: Any,
converter_cls: Optional[Type[Converter]] = None,
) -> Union[dict, BaseModel, str]:
print("convert_with_instructions")
llm = agent.function_calling_llm or agent.llm
print("llm", llm)
instructions = get_conversion_instructions(model, llm)
print("instructions", instructions)
converter = create_converter(
agent=agent,
converter_cls=converter_cls,
@@ -174,11 +170,9 @@ def convert_with_instructions(
model=model,
instructions=instructions,
)
print("converter", converter)
exported_result = (
converter.to_pydantic() if not is_json_output else converter.to_json()
)
print("exported_result", exported_result)
if isinstance(exported_result, ConverterError):
Printer().print(