diff --git a/aios/hooks/stores/processes.py b/aios/hooks/stores/processes.py index dfb6fbbb..5385fafa 100644 --- a/aios/hooks/stores/processes.py +++ b/aios/hooks/stores/processes.py @@ -10,3 +10,5 @@ def addProcess(p: Future, pi: str) -> None: def clearProcesses() -> None: AGENT_PROCESSES.clear() + +def addId diff --git a/aios/llm_core/cores/api/google.py b/aios/llm_core/cores/api/google.py index ce5a42d2..6fe8ea7c 100644 --- a/aios/llm_core/cores/api/google.py +++ b/aios/llm_core/cores/api/google.py @@ -8,7 +8,7 @@ from aios.utils.utils import get_from_env -from pyopenagi.utils.chat_template import Response +from cerebrum.llm.communication import Response class GeminiLLM(BaseLLM):