Former-commit-id: bba18969ba98702a2e9b6e67b9ef1e167cb043f3
pull/160/head
Kye 2 years ago
parent 26e384bed1
commit 15ad97b274

@ -1,10 +1,10 @@
#aug 10 #aug 10
#Vortex is the name of my Duck friend, ILY Vortex #Vortex is the name of my Duck pet, ILY Vortex
#Kye #Kye
import logging import logging
import os import os
from typing import List, Optional, Union from typing import List, Optional, Union, Any
import faiss import faiss
from langchain.agents import Tool from langchain.agents import Tool
@ -36,7 +36,7 @@ class VortexWorkerAgent:
def __init__(self, def __init__(self,
openai_api_key: str, openai_api_key: str,
llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None, llm: Optional[Union[InMemoryDocstore, ChatOpenAI]] = None,
tools: Optional[List[Tool]] = None, tools: Optional[Any] = None,
embedding_size: Optional[int] = 8192, embedding_size: Optional[int] = 8192,
worker_name: Optional[str] = "Vortex Worker Agent", worker_name: Optional[str] = "Vortex Worker Agent",
worker_role: Optional[str] = "Assistant", worker_role: Optional[str] = "Assistant",

Loading…
Cancel
Save