From aa2354de5c0dfe3f2443a74d45a3bcb43154d1e7 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 10 Aug 2023 11:22:43 -0400 Subject: [PATCH] prompt logic Former-commit-id: c0d9261f13fcd228b9b9ee3085eff0361f1c5571 --- swarms/agents/base.py | 2 +- .../prompts/{prompt_generator.py => agent_prompt_generator.py} | 0 swarms/workers/vortex_worker.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename swarms/agents/models/prompts/{prompt_generator.py => agent_prompt_generator.py} (100%) diff --git a/swarms/agents/base.py b/swarms/agents/base.py index fd67abc2..6c7122d6 100644 --- a/swarms/agents/base.py +++ b/swarms/agents/base.py @@ -20,7 +20,7 @@ from swarms.agents.models.prompts.agent_prompt_auto import ( MessageFormatter, PromptConstructor, ) -from swarms.agents.models.prompts.prompt_generator import FINISH_NAME +from swarms.agents.models.prompts.agent_prompt_generator import FINISH_NAME from swarms.agents.tools.base import BaseTool from swarms.agents.utils.Agent import AgentOutputParser from swarms.agents.utils.human_input import HumanInputRun diff --git a/swarms/agents/models/prompts/prompt_generator.py b/swarms/agents/models/prompts/agent_prompt_generator.py similarity index 100% rename from swarms/agents/models/prompts/prompt_generator.py rename to swarms/agents/models/prompts/agent_prompt_generator.py diff --git a/swarms/workers/vortex_worker.py b/swarms/workers/vortex_worker.py index 2c64243b..95293cb4 100644 --- a/swarms/workers/vortex_worker.py +++ b/swarms/workers/vortex_worker.py @@ -87,7 +87,7 @@ class VortexWorkerAgent: try: openai_api_key = self.openai_api_key or os.getenv("OPENAI_API_KEY") embeddings_model = OpenAIEmbeddings(openai_api_key=openai_api_key) - index = faiss.IndexFlatL2(self.embedding_size) + index = faiss.IndexFlatL2(8192) return FAISS(embeddings_model, index, InMemoryDocstore({}), {}) except Exception as error: logging.error(f"Failed to initialize vector store: {error}")