7|h6ddlmZddlmZmZmZddlZddlmZddl m Z Gdde Z Gdd e Z Gd d e Z Gd d e Zy)) annotations)AnyDictOptionalN)CallbackManagerForToolRun)BaseToolcxeZdZUdZdZded<dZded<ded<ded<ded <d Zded < d dd ZddZ y )CogniswitchKnowledgeRequesta!Tool that uses the Cogniswitch service to answer questions. name: str = "cogniswitch_knowledge_request" description: str = ( "A wrapper around cogniswitch service to answer the question from the knowledge base." "Input should be a search query." ) cogniswitch_knowledge_requeststrnamezYA wrapper around cogniswitch service to answer the question from the knowledge base. descriptioncs_token OAI_tokenapiKeyz@https://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeRequestapi_urlNcj|j|j|j||j}|S)a Use the tool to answer a query. Args: query (str): Natural language query, that you would like to ask to your knowledge graph. run_manager (Optional[CallbackManagerForChainRun]): Manager for chain run callbacks. Returns: Dict[str, Any]: Output dictionary containing the 'response' from the service. ) answer_csrrr)selfquery run_managerresponses i/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/tools/cogniswitch/tool.py_runz CogniswitchKnowledgeRequest._runs)$>>$-- Tc|s td|s td|s td|s td|||d}d|i}tj|j|d|}|j S) ac Send a query to the Cogniswitch service and retrieve the response. Args: cs_token (str): Cogniswitch token. OAI_token (str): OpenAI token. apiKey (str): OAuth token. query (str): Query to be answered. Returns: dict: Response JSON from the Cogniswitch service. zMissing cs_tokenzMissing OpenAI tokenzMissing cogniswitch OAuth tokenzMissing input query)r platformToken openAITokenrFheadersverifydata) ValueErrorrequestspostrjson)rrrrrr r"rs rrz%CogniswitchKnowledgeRequest.answer_cs2s/0 034 4>? ?23 3%$  ==wuSWX}}rN)rr r#Optional[CallbackManagerForToolRun]returnDict[str, Any]) rr rr rr rr r)dict) __name__ __module__ __qualname____doc__r __annotations__rrrrrrr r sf0D#/4K4MN KUGSU <@9  *rr cxeZdZUdZdZded<dZded<ded<ded<ded <d Zded < d dd ZddZ y )CogniswitchKnowledgeStatusaeTool that uses the Cogniswitch services to get the status of the document or url uploaded. name: str = "cogniswitch_knowledge_status" description: str = ( "A wrapper around cogniswitch services to know the status of the document uploaded from a url or a file. " "Input should be a file name or the url link" ) cogniswitch_knowledge_statusr r zlA wrapper around cogniswitch services to know the status of the document uploaded from a url or a file.rrrrzFhttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/statusknowledge_status_urlNc(|j|}|S)a Use the tool to know the status of the document uploaded. Args: document_name (str): name of the document or the url uploaded run_manager (Optional[CallbackManagerForChainRun]): Manager for chain run callbacks. Returns: Dict[str, Any]: Output dictionary containing the 'response' from the service. )knowledge_status)r document_namerrs rrzCogniswitchKnowledgeStatus._runis$((7rc,||jd}|j|j|jd}tj|j ||d}|j dk(r!|j}t|d}|j d}|dk(rd |d<n6|d k(rd |d<n+|d k(rd |d<n |dk(rd|d<n|dk(rd|d<n |dk(rd|d<d|jvr|jdd|jvr|jdd|jvr|jdd|jvr|jdd|jvr|jd|Sd|j iS)a Use this function to know the status of the document or the URL uploaded Args: document_name (str): The document name or the url that is uploaded. Returns: dict: Response JSON from the Cogniswitch service. )docNamerrrrF)r paramsr!statusrSUCCESS PROCESSINGUPLOADEDFAILUREUPLOAD_FAILUREREJECTEDfilePath savedFileNameintegrationConfigIdmetaData docEntryIdmessage) rrrr$getr5 status_coder&r+keyspop)rr8r<r r source_info source_datar?s rr7z+CogniswitchKnowledgeStatus.knowledge_status~s+T]]Kkk>>!]]  <<  % %     3 &"--/K{2/K __X.F{(1 H%1(4 H%1(2 H%1(1 H%1(8 H%1(2 H%[--// ++"2"2"440$ (8(8(:: 56[--// +{//11 - 8// rr')r8r rr(r)r*)r8r r)r+) r,r-r.r/r r0rr5rr7r1rrr3r3Ssm /D#.AKAMN KP#<@9  *6rr3ceZdZUdZdZded<dZded<ded<ded<ded <d Zded < d dd Z ddZ y )CogniswitchKnowledgeSourceFileaTool that uses the Cogniswitch services to store data from file. name: str = "cogniswitch_knowledge_source_file" description: str = ( "This calls the CogniSwitch services to analyze & store data from a file. If the input looks like a file path, assign that string value to file key. Assign document name & description only if provided in input." ) !cogniswitch_knowledge_source_filer r z This calls the CogniSwitch services to analyze & store data from a file. If the input looks like a file path, assign that string value to file key. Assign document name & description only if provided in input. rrrrzDhttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/fileknowledgesource_fileNc:|sddiS|j|||}|S)a Execute the tool to store the data given from a file. This calls the CogniSwitch services to analyze & store data from a file. If the input looks like a file path, assign that string value to file key. Assign document name & description only if provided in input. Args: file Optional[str]: The file path of your knowledge document_name Optional[str]: Name of your knowledge document document_description Optional[str]: Description of your knowledge document run_manager (Optional[CallbackManagerForChainRun]): Manager for chain run callbacks. Returns: Dict[str, Any]: Output dictionary containing the 'response' from the service. rPNo input provided)filer8document_description store_data)rr]r8r^rrs rrz#CogniswitchKnowledgeSourceFile._runs<0. +%9'H Orc|j|j|jd}|sd}|sd}|dt|di}||d}t j |j |d|}|jdk(r|jSd d iS) a Store data using the Cogniswitch service. This calls the CogniSwitch services to analyze & store data from a file. If the input looks like a file path, assign that string value to file key. Assign document name & description only if provided in input. Args: file (Optional[str]): file path of your file. the current files supported by the files are .txt, .pdf, .docx, .doc, .html document_name (Optional[str]): Name of the document you are uploading. document_description (Optional[str]): Description of the document. Returns: dict: Response JSON from the Cogniswitch service. r;r]rb) documentNamedocumentDescriptionF)r r!r"filesr=rP Bad Request) rrropenr$r%rZrRr&)rr]r8r^r rfr"rs rr`z)CogniswitchKnowledgeSourceFile.store_datas.kk>>!]]  M##%  T$-.E*#7 ==  % %     3 &==? "}- -rNNNN) r] Optional[str]r8rjr^rjrr(r)r*)r]rjr8rjr^rjr)r+) r,r-r.r/r r0rrZrr`r1rrrXrXs4D#3 K MN KN# #'+.2;? ""%", " 9 "  "H2.2.%2., 2.  2.rrXceZdZUdZdZded<dZded<ded<ded<ded <d Zded < d dd Z ddZ y )CogniswitchKnowledgeSourceURLaxTool that uses the Cogniswitch services to store data from a URL. name: str = "cogniswitch_knowledge_source_url" description: str = ( "This calls the CogniSwitch services to analyze & store data from a url. the URL is provided in input, assign that value to the url key. Assign document name & description only if provided in input" ) cogniswitch_knowledge_source_urlr r z This calls the CogniSwitch services to analyze & store data from a url. the URL is provided in input, assign that value to the url key. Assign document name & description only if provided in inputrrrrzChttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/urlknowledgesource_urlNc:|sddiS|j|||}|S)a Execute the tool to store the data given from a url. This calls the CogniSwitch services to analyze & store data from a url. the URL is provided in input, assign that value to the url key. Assign document name & description only if provided in input. Args: url Optional[str]: The website/url link of your knowledge document_name Optional[str]: Name of your knowledge document document_description Optional[str]: Description of your knowledge document run_manager (Optional[CallbackManagerForChainRun]): Manager for chain run callbacks. Returns: Dict[str, Any]: Output dictionary containing the 'response' from the service. rPr\)urlr8r^r_)rrpr8r^rrs rrz"CogniswitchKnowledgeSourceURL._run?s<0. ??'!5#  rc|j|j|jd}|sd}|sd}|sddiSd|i}tj|j |d|}|j dk(r|jSdd iS) aE Store data using the Cogniswitch service. This calls the CogniSwitch services to analyze & store data from a url. the URL is provided in input, assign that value to the url key. Assign document name & description only if provided in input. Args: url (Optional[str]): URL link. document_name (Optional[str]): Name of the document you are uploading. document_description (Optional[str]): Description of the document. Returns: dict: Response JSON from the Cogniswitch service. r;rbrPr\rpFrr=rg)rrrr$r%rnrRr&)rrpr8r^r r"rs rr`z(CogniswitchKnowledgeSourceURL.store_databs*kk>>!]]  M##% . 3rvsR"&&>)F(FRaaHn.Xn.bg.Hg.r