mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-09-30 15:50:23 +03:00
Merge pull request #2910 from markparonyan/mypy-agent-copilot-utils
fix(typing): resolve mypy errors in agent.gns3_copilot.utils
This commit is contained in:
commit
4bcaf596df
@ -37,6 +37,7 @@ from typing import Dict
|
|||||||
from langchain_core.messages import AIMessage
|
from langchain_core.messages import AIMessage
|
||||||
from langchain_core.messages import HumanMessage
|
from langchain_core.messages import HumanMessage
|
||||||
from langchain_core.messages import SystemMessage
|
from langchain_core.messages import SystemMessage
|
||||||
|
from langchain_core.messages import ToolCall
|
||||||
from langchain_core.messages import ToolMessage
|
from langchain_core.messages import ToolMessage
|
||||||
|
|
||||||
|
|
||||||
@ -86,7 +87,7 @@ def convert_langchain_to_openai(lc_message) -> Dict[str, Any]:
|
|||||||
tool_calls = []
|
tool_calls = []
|
||||||
for tc in lc_message.tool_calls:
|
for tc in lc_message.tool_calls:
|
||||||
# Convert to dict if it's an object
|
# Convert to dict if it's an object
|
||||||
tc_dict = tc if isinstance(tc, dict) else tc.model_dump()
|
tc_dict = tc
|
||||||
tool_calls.append(
|
tool_calls.append(
|
||||||
{
|
{
|
||||||
"id": tc_dict.get("id", str(uuid.uuid4())),
|
"id": tc_dict.get("id", str(uuid.uuid4())),
|
||||||
@ -139,7 +140,7 @@ def convert_openai_to_langchain(msg: Dict[str, Any]):
|
|||||||
|
|
||||||
# Restore tool calls if present
|
# Restore tool calls if present
|
||||||
if "tool_calls" in msg and msg["tool_calls"]:
|
if "tool_calls" in msg and msg["tool_calls"]:
|
||||||
tool_calls = []
|
tool_calls: list[ToolCall] = []
|
||||||
for tc in msg["tool_calls"]:
|
for tc in msg["tool_calls"]:
|
||||||
tool_calls.append(
|
tool_calls.append(
|
||||||
{
|
{
|
||||||
|
|||||||
@ -214,9 +214,7 @@ def parse_tool_content(
|
|||||||
return content
|
return content
|
||||||
|
|
||||||
# Handle unsupported types
|
# Handle unsupported types
|
||||||
error_msg = ( # type: ignore[unreachable]
|
error_msg = f"Content must be str, dict, list, int, float, bool, or None, got {type(content).__name__}"
|
||||||
f"Content must be str, dict, list, int, float, bool, or None, got {type(content).__name__}"
|
|
||||||
)
|
|
||||||
logger.error(error_msg)
|
logger.error(error_msg)
|
||||||
|
|
||||||
if strict_mode:
|
if strict_mode:
|
||||||
|
|||||||
@ -301,8 +301,6 @@ module = [
|
|||||||
"gns3server.agent.gns3_copilot.tools_v2.gns3_stop_node", # 3
|
"gns3server.agent.gns3_copilot.tools_v2.gns3_stop_node", # 3
|
||||||
"gns3server.agent.gns3_copilot.tools_v2.gns3_suspend_node", # 3
|
"gns3server.agent.gns3_copilot.tools_v2.gns3_suspend_node", # 3
|
||||||
"gns3server.agent.gns3_copilot.utils.custom_netmiko.scripts.list_netmiko_telnet_devices", # 1
|
"gns3server.agent.gns3_copilot.utils.custom_netmiko.scripts.list_netmiko_telnet_devices", # 1
|
||||||
"gns3server.agent.gns3_copilot.utils.message_converters", # 2
|
|
||||||
"gns3server.agent.gns3_copilot.utils.parse_tool_content", # 1
|
|
||||||
"gns3server.agent.mcp.appliances", # 1
|
"gns3server.agent.mcp.appliances", # 1
|
||||||
"gns3server.agent.web_wireshark.docker_client", # 5
|
"gns3server.agent.web_wireshark.docker_client", # 5
|
||||||
"gns3server.api.routes.compute.atm_switch_nodes", # 9
|
"gns3server.api.routes.compute.atm_switch_nodes", # 9
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user