mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-10-01 00:02:34 +03:00
fix(typing): resolve mypy errors in agent.gns3_copilot.utils.message_converters
This commit is contained in:
parent
b9192438c9
commit
5cb1b8e3d2
@ -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(
|
||||||
{
|
{
|
||||||
|
|||||||
@ -312,7 +312,6 @@ module = [
|
|||||||
"gns3server.agent.gns3_copilot.tools_v2.packet_analysis_tool", # 1
|
"gns3server.agent.gns3_copilot.tools_v2.packet_analysis_tool", # 1
|
||||||
"gns3server.agent.gns3_copilot.tools_v2.vpcs_tools_netmiko", # 9
|
"gns3server.agent.gns3_copilot.tools_v2.vpcs_tools_netmiko", # 9
|
||||||
"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.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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user