Ng6ddlmZddlmZmZmZddlZddlmZddl m Z Gdde Z Gdd e Z Gd d e Z Gd d e ZdS)) annotations)AnyDictOptionalN)CallbackManagerForToolRun)BaseToolcpeZdZUdZdZded<dZded<ded<ded<ded <d Zded < dddZddZ d S)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_urlNquery run_manager#Optional[CallbackManagerForToolRun]returnDict[str, Any]cT||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)selfrrresponses f/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/tools/cogniswitch/tool.py_runz CogniswitchKnowledgeRequest._runs&$>>$- TTdictc|std|std|std|std|||d}d|i}tj|j|d|}|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 1/00 0 5344 4 @>?? ? 4233 3%$   =wuSWXXX}}rN)rr rrrr) rr rr rr rr rr) __name__ __module__ __qualname____doc__r __annotations__rrrrrrr r s0D////4K4444MMMNNNKKKUGUUUU <@*rr cpeZdZUdZdZded<dZded<ded<ded<ded <d Zded < dddZddZ d S)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_urlN document_namerrrrc0||}|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)rr6rrs rrzCogniswitchKnowledgeStatus._runis$((77rrc`||jd}|j|j|jd}tj|j||d}|jdkr^|}t|d}|d}|dkrd |d<n;|d krd |d<n/|d krd |d<n#|dkrd|d<n|dkrd|d<n |dkrd|d<d| vr| dd| vr| dd| vr| dd| vr| dd| vr| d|Sd|jiS)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. )docNamer!rr"r!F)r$paramsr%statusrSUCCESS PROCESSINGUPLOADEDFAILUREUPLOAD_FAILUREREJECTEDfilePath savedFileNameintegrationConfigIdmetaData docEntryIdmessage) rrrr(getr5 status_coder*rkeyspop)rr6r<r$r source_info source_datar?s rr8z+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"4"444000$ (8(8(:(::: 5666[--//// +++{//1111 --- 8/ rr+)r6r rrrr)r6r rr) r,r-r.r/r r0rr5rr8r1rrr3r3Ss  /D....AKAAAAMMMNNNKKKP<@*666666rr3cveZdZUdZdZded<dZded<ded<ded<ded <d Zded < dddZddZ d S)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_fileNfile Optional[str]r6document_descriptionrrrrcB|sddiS||||}|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)r[r6r] store_data)rr[r6r]rrs rrz#CogniswitchKnowledgeSourceFile._runsF0 . +%9'H Orrc|j|j|jd}|sd}|sd}|dt|di}||d}t j|j|d||}|jd kr|Sd 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;Nr[rb) documentNamedocumentDescriptionF)r$r%r&filesr=rP Bad Request) rrropenr(r)rZrRr*)rr[r6r]r$rgr&rs rraz)CogniswitchKnowledgeSourceFile.store_datas.k>!]    M# &#%  T$--.E*#7  =  %      3 & &==?? "}- -rNNNN) r[r\r6r\r]r\rrrr)r[r\r6r\r]r\rr) r,r-r.r/r r0rrZrrar1rrrXrXs4D3333 K    MMMNNNKKKN #'+.2;? """""H2.2.2.2.2.2.rrXcveZdZUdZdZded<dZded<ded<ded<ded <d Zded < dddZddZ d S)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_urlNurlr\r6r]rrrrcB|sddiS||||}|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_)ror6r]r`)rror6r]rrs rrz"CogniswitchKnowledgeSourceURL._run?sF0 . ??'!5#   rrc|j|j|jd}|sd}|sd}|sddiSd|i}tj|j|d|}|jdkr|Sdd 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;rcrPr_roFr#r=rh)rrrr(r)rnrRr*)rror6r]r$r&rs rraz(CogniswitchKnowledgeSourceURL.store_databs*k>!]    M# &#%  . 3rvs9""""""&&&&&&&&&&>>>>>>))))))FFFFF(FFFRaaaaaaaaHn.n.n.n.n.Xn.n.n.bg.g.g.g.g.Hg.g.g.g.g.r