mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-10-25 03:12:23 +02:00
Merge pull request #2905 from markparonyan/mypy-agent-copilot-agent
fix(typing): resolve mypy errors in agent.gns3_copilot.agent
This commit is contained in:
commit
0ef6e80d94
@ -140,7 +140,7 @@ def estimate_tool_tokens(tools: list[Any]) -> int:
|
|||||||
for tool in tools:
|
for tool in tools:
|
||||||
try:
|
try:
|
||||||
# Build tool schema in OpenAI format
|
# Build tool schema in OpenAI format
|
||||||
tool_schema = {
|
tool_schema: dict[str, Any] = {
|
||||||
"type": "function",
|
"type": "function",
|
||||||
"function": {
|
"function": {
|
||||||
"name": tool.name,
|
"name": tool.name,
|
||||||
|
|||||||
@ -110,7 +110,7 @@ def create_base_model(
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
# Prepare parameters for init_chat_model
|
# Prepare parameters for init_chat_model
|
||||||
init_params = {
|
init_params: dict[str, Any] = {
|
||||||
"model": config_vars["model_name"],
|
"model": config_vars["model_name"],
|
||||||
"model_provider": config_vars["model_provider"],
|
"model_provider": config_vars["model_provider"],
|
||||||
"api_key": config_vars["api_key"],
|
"api_key": config_vars["api_key"],
|
||||||
@ -177,7 +177,7 @@ def create_title_model(
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
# Prepare parameters for init_chat_model
|
# Prepare parameters for init_chat_model
|
||||||
init_params = {
|
init_params: dict[str, Any] = {
|
||||||
"model": config_vars["model_name"],
|
"model": config_vars["model_name"],
|
||||||
"model_provider": config_vars["model_provider"],
|
"model_provider": config_vars["model_provider"],
|
||||||
"api_key": config_vars["api_key"],
|
"api_key": config_vars["api_key"],
|
||||||
|
|||||||
@ -295,8 +295,6 @@ enable_error_code = ["ignore-without-code", "redundant-expr", "truthy-bool"]
|
|||||||
# Remove modules from this list one small PR at a time. Never add new ones.
|
# Remove modules from this list one small PR at a time. Never add new ones.
|
||||||
module = [
|
module = [
|
||||||
"gns3server.agent", # 2
|
"gns3server.agent", # 2
|
||||||
"gns3server.agent.gns3_copilot.agent.context_manager", # 5
|
|
||||||
"gns3server.agent.gns3_copilot.agent.model_factory", # 4
|
|
||||||
"gns3server.agent.gns3_copilot.chat_sessions_repository", # 8
|
"gns3server.agent.gns3_copilot.chat_sessions_repository", # 8
|
||||||
"gns3server.agent.gns3_copilot.gns3_client.api_handlers", # 10
|
"gns3server.agent.gns3_copilot.gns3_client.api_handlers", # 10
|
||||||
"gns3server.agent.gns3_copilot.gns3_client.connector_factory", # 1
|
"gns3server.agent.gns3_copilot.gns3_client.connector_factory", # 1
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user