prompt logic

Former-commit-id: f0634daf5a
pull/47/head
Kye 1 year ago
parent 4878c981dc
commit 4c46debe1e

@ -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

@ -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}")

Loading…
Cancel
Save