refactor: make sure that in main the correct names and passwords are called for starting the agents
ref: N25B-257
This commit is contained in:
4
src/control_backend/agents/bel_agents/__init__.py
Normal file
4
src/control_backend/agents/bel_agents/__init__.py
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
from .bel_collector_agent.bel_collector_agent import BelCollectorAgent as BelCollectorAgent
|
||||||
|
from .bel_text_extract_agent.bel_text_extract_agent import (
|
||||||
|
BelTextExtractAgent as BelTextExtractAgent,
|
||||||
|
)
|
||||||
@@ -1 +0,0 @@
|
|||||||
from .bel_collector_agent import BelCollectorAgent as BelCollectorAgent
|
|
||||||
@@ -11,7 +11,7 @@ class AgentSettings(BaseModel):
|
|||||||
host: str = "localhost"
|
host: str = "localhost"
|
||||||
bdi_core_agent_agent_name: str = "bdi_core_agent"
|
bdi_core_agent_agent_name: str = "bdi_core_agent"
|
||||||
bel_collector_agent_name: str = "bel_collector_agent"
|
bel_collector_agent_name: str = "bel_collector_agent"
|
||||||
texbel_text_extractor_agent_name: str = "text_belief_extractor"
|
bel_text_extractor_agent_name: str = "bel_text_extractor_agent"
|
||||||
per_vad_agent_name: str = "per_vad_agent"
|
per_vad_agent_name: str = "per_vad_agent"
|
||||||
llm_agent_name: str = "llm_agent"
|
llm_agent_name: str = "llm_agent"
|
||||||
test_agent_name: str = "test_agent"
|
test_agent_name: str = "test_agent"
|
||||||
|
|||||||
@@ -7,13 +7,24 @@ from fastapi import FastAPI
|
|||||||
from fastapi.middleware.cors import CORSMiddleware
|
from fastapi.middleware.cors import CORSMiddleware
|
||||||
from zmq.asyncio import Context
|
from zmq.asyncio import Context
|
||||||
|
|
||||||
from control_backend.agents import (
|
# Act agents
|
||||||
BelCollectorAgent,
|
# BDI agents
|
||||||
ComRIAgent,
|
from control_backend.agents.bdi_agents import BDICoreAgent
|
||||||
LLMAgent,
|
|
||||||
PerVADAgent,
|
# Believe Agents
|
||||||
)
|
from control_backend.agents.bel_agents import BelCollectorAgent, BelTextExtractAgent
|
||||||
from control_backend.agents.bdi_agents import BDICoreAgent, BelTextExtractAgent
|
|
||||||
|
# Communication agents
|
||||||
|
from control_backend.agents.com_agents import ComRIAgent
|
||||||
|
|
||||||
|
# Emotional Agents
|
||||||
|
# LLM Agents
|
||||||
|
from control_backend.agents.llm_agents import LLMAgent
|
||||||
|
|
||||||
|
# Perceive agents
|
||||||
|
from control_backend.agents.per_agents import PerVADAgent
|
||||||
|
|
||||||
|
# Other backend imports
|
||||||
from control_backend.api.v1.router import api_router
|
from control_backend.api.v1.router import api_router
|
||||||
from control_backend.core.config import settings
|
from control_backend.core.config import settings
|
||||||
from control_backend.logging import setup_logging
|
from control_backend.logging import setup_logging
|
||||||
@@ -105,10 +116,10 @@ async def lifespan(app: FastAPI):
|
|||||||
"TBeliefExtractor": (
|
"TBeliefExtractor": (
|
||||||
BelTextExtractAgent,
|
BelTextExtractAgent,
|
||||||
{
|
{
|
||||||
"name": settings.agent_settings.texbel_text_extractor_agent_name,
|
"name": settings.agent_settings.bel_text_extractor_agent_name,
|
||||||
"jid": f"{settings.agent_settings.texbel_text_extractor_agent_name}@"
|
"jid": f"{settings.agent_settings.bel_text_extractor_agent_name}@"
|
||||||
f"{settings.agent_settings.host}",
|
f"{settings.agent_settings.host}",
|
||||||
"password": settings.agent_settings.texbel_text_extractor_agent_name,
|
"password": settings.agent_settings.bel_text_extractor_agent_name,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
"PerVADAgent": (
|
"PerVADAgent": (
|
||||||
|
|||||||
Reference in New Issue
Block a user