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"
|
||||
bdi_core_agent_agent_name: str = "bdi_core_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"
|
||||
llm_agent_name: str = "llm_agent"
|
||||
test_agent_name: str = "test_agent"
|
||||
|
||||
@@ -7,13 +7,24 @@ from fastapi import FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from zmq.asyncio import Context
|
||||
|
||||
from control_backend.agents import (
|
||||
BelCollectorAgent,
|
||||
ComRIAgent,
|
||||
LLMAgent,
|
||||
PerVADAgent,
|
||||
)
|
||||
from control_backend.agents.bdi_agents import BDICoreAgent, BelTextExtractAgent
|
||||
# Act agents
|
||||
# BDI agents
|
||||
from control_backend.agents.bdi_agents import BDICoreAgent
|
||||
|
||||
# Believe Agents
|
||||
from control_backend.agents.bel_agents import BelCollectorAgent, 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.core.config import settings
|
||||
from control_backend.logging import setup_logging
|
||||
@@ -105,10 +116,10 @@ async def lifespan(app: FastAPI):
|
||||
"TBeliefExtractor": (
|
||||
BelTextExtractAgent,
|
||||
{
|
||||
"name": settings.agent_settings.texbel_text_extractor_agent_name,
|
||||
"jid": f"{settings.agent_settings.texbel_text_extractor_agent_name}@"
|
||||
"name": settings.agent_settings.bel_text_extractor_agent_name,
|
||||
"jid": f"{settings.agent_settings.bel_text_extractor_agent_name}@"
|
||||
f"{settings.agent_settings.host}",
|
||||
"password": settings.agent_settings.texbel_text_extractor_agent_name,
|
||||
"password": settings.agent_settings.bel_text_extractor_agent_name,
|
||||
},
|
||||
),
|
||||
"PerVADAgent": (
|
||||
|
||||
Reference in New Issue
Block a user