chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
import logging
|
||||
|
||||
from open_chatcaht._constants import API_BASE_URI
|
||||
from open_chatcaht.api.chat.chat_client import ChatClient
|
||||
from open_chatcaht.api.knowledge_base.knowledge_base_client import KbClient
|
||||
from open_chatcaht.api.server.server_client import ServerClient
|
||||
from open_chatcaht.api.standard_openai.standard_openai_client import StandardOpenaiClient
|
||||
from open_chatcaht.api.tools.tool_client import ToolClient
|
||||
|
||||
|
||||
class ChatChat:
|
||||
knowledge: KbClient = None
|
||||
tool: ToolClient = None
|
||||
server: ServerClient = None
|
||||
chat: ChatClient = None
|
||||
openai_adapter: StandardOpenaiClient = None
|
||||
|
||||
def __init__(self,
|
||||
base_url: str = API_BASE_URI,
|
||||
timeout: float = 60,
|
||||
use_async: bool = False,
|
||||
use_proxy: bool = False,
|
||||
proxies=None,
|
||||
log_level: int = logging.INFO,
|
||||
retry: int = 3,
|
||||
retry_interval: int = 1, ):
|
||||
param = {
|
||||
'log_level': log_level,
|
||||
'retry': retry,
|
||||
'retry_interval': retry_interval,
|
||||
'base_url': base_url,
|
||||
'timeout': timeout,
|
||||
'use_async': use_async,
|
||||
'use_proxy': use_proxy,
|
||||
'proxies': proxies
|
||||
}
|
||||
|
||||
self.knowledge = KbClient(**param)
|
||||
self.tool = ToolClient(**param)
|
||||
self.server = ServerClient(**param)
|
||||
self.chat = ChatClient(**param)
|
||||
self.openai_adapter = StandardOpenaiClient(**param)
|
||||
|
||||
Reference in New Issue
Block a user