pull/48/head
Kye 1 year ago
parent 53ed7aa9c7
commit 95ccc45dbf

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