a bg6@sddlmZddlmZmZmZddlZddlmZddl m Z Gddde Z Gdd d e Z Gd d d e Z Gd d d e ZdS)) annotations)AnyDictOptionalN)CallbackManagerForToolRun)BaseToolc@sxeZdZUdZdZded<dZded<ded<ded<ded <d Zded <ddd ddddZddddddddZ 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." ) Zcogniswitch_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_urlN#Optional[CallbackManagerForToolRun]Dict[str, Any])query run_managerreturncCs||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_csr r r)selfrrresponserx/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/cogniswitch/tool.py_runsz CogniswitchKnowledgeRequest._rundict)r r rrrcCs`|s td|std|s$td|s0td|||d}d|i}tj|j|d|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)rr r rrrr!rrrrr2s z%CogniswitchKnowledgeRequest.answer_cs)N) __name__ __module__ __qualname____doc__r __annotations__r rrrrrrrr s    rc@sreZdZUdZdZded<dZded<ded<ded<ded <d Zded <ddd ddddZdddddZ 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" ) Zcogniswitch_knowledge_statusr r zlA wrapper around cogniswitch services to know the status of the document uploaded from a url or a file.r r r rzFhttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/statusknowledge_status_urlNrr) document_namerrcCs||}|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)rr-rrrrrris zCogniswitchKnowledgeStatus._runr)r-rcCsL||jd}|j|j|jd}tj|j||dd}|jdkr>|}t|d}|d}|dkrnd |d<nX|d krd |d<nF|d krd |d<n4|dkrd|d<n"|dkrd|d<n|dkrd|d<d| vr| dd| vr| dd| vr | dd| vr"| dd| vr:| d|Sd|jiSdS)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. )ZdocNamerrrrF)rparamsr statusrSUCCESS PROCESSINGZUPLOADEDFAILUREZUPLOAD_FAILUREZREJECTEDZfilePathZ savedFileNameZintegrationConfigIdZmetaDataZ docEntryIdmessageN) r rr r#getr, status_coder%rkeyspop)rr-r0rrZ source_infoZ source_datar3rrrr.~sP                z+CogniswitchKnowledgeStatus.knowledge_status)N) r&r'r(r)r r*r r,rr.rrrrr+Ss    r+c@szeZdZUdZdZded<dZded<ded<ded<ded <d Zded <dd d d dddddZd d d ddddZ 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." ) Z!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. r r r rzDhttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/fileknowledgesource_fileN Optional[str]rr)filer-document_descriptionrrcCs$|s ddiS|j|||d}|SdS)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. r<No input provided)rDr-rEN store_data)rrDr-rErrrrrrsz#CogniswitchKnowledgeSourceFile._runr)rDr-rErcCsv|j|j|jd}|sd}|s"d}|dur8dt|di}||d}tj|j|d||d}|jd krj|Sd d iSdS) 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/NrDrb)Z documentNameZdocumentDescriptionF)rr r!filesr1r< Bad Request) rr r openr#r$rBr>r%)rrDr-rErrKr!rrrrrHs. z)CogniswitchKnowledgeSourceFile.store_data)NNNN) r&r'r(r)r r*r rBrrHrrrrrAs    $rAc@szeZdZUdZdZded<dZded<ded<ded<ded <d Zded <dd d d dddddZd d d ddddZ 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" ) Z 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 inputr r r rzChttps://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/urlknowledgesource_urlNrCrr)urlr-rErrcCs |s ddiS|j|||d}|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. r<rF)rPr-rErG)rrPr-rErrrrrr?sz"CogniswitchKnowledgeSourceURL._runr)rPr-rErcCsh|j|j|jd}|sd}|s"d}|s.ddiSd|i}tj|j|d|d}|jdkr\|Sdd iSd S) 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/rIr<rFrPFrr1rLN)rr r r#r$rOr>r%)rrPr-rErr!rrrrrHbs* z(CogniswitchKnowledgeSourceURL.store_data)NNNN) r&r'r(r)r r*r rOrrHrrrrrN(s    #rN) __future__rtypingrrrr#Zlangchain_core.callbacksrZlangchain_core.toolsrrr+rArNrrrrs   Idq