Source code for langchain_community.tools.cogniswitch.tool

from __future__ import annotations

from typing import Any, Dict, Optional

import requests
from langchain_core.callbacks import CallbackManagerForToolRun
from langchain_core.tools import BaseTool


[docs]class CogniswitchKnowledgeRequest(BaseTool): """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." ) """ name: str = "cogniswitch_knowledge_request" description: str = """A wrapper around cogniswitch service to answer the question from the knowledge base.""" cs_token: str OAI_token: str apiKey: str api_url: str = "https://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeRequest" def _run( self, query: str, run_manager: Optional[CallbackManagerForToolRun] = None, ) -> Dict[str, Any]: """ 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. """ response = self.answer_cs(self.cs_token, self.OAI_token, query, self.apiKey) return response
[docs] def answer_cs(self, cs_token: str, OAI_token: str, query: str, apiKey: str) -> dict: """ 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. """ if not cs_token: raise ValueError("Missing cs_token") if not OAI_token: raise ValueError("Missing OpenAI token") if not apiKey: raise ValueError("Missing cogniswitch OAuth token") if not query: raise ValueError("Missing input query") headers = { "apiKey": apiKey, "platformToken": cs_token, "openAIToken": OAI_token, } data = {"query": query} response = requests.post(self.api_url, headers=headers, verify=False, data=data) return response.json()
[docs]class CogniswitchKnowledgeStatus(BaseTool): """Tool 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" ) """ 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.""" cs_token: str OAI_token: str apiKey: str knowledge_status_url = ( "https://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/status" ) def _run( self, document_name: str, run_manager: Optional[CallbackManagerForToolRun] = None, ) -> Dict[str, Any]: """ 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. """ response = self.knowledge_status(document_name) return response
[docs] def knowledge_status(self, document_name: str) -> dict: """ 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. """ params = {"docName": document_name, "platformToken": self.cs_token} headers = { "apiKey": self.apiKey, "openAIToken": self.OAI_token, "platformToken": self.cs_token, } response = requests.get( self.knowledge_status_url, headers=headers, params=params, verify=False, ) if response.status_code == 200: source_info = response.json() source_data = dict(source_info[-1]) status = source_data.get("status") if status == 0: source_data["status"] = "SUCCESS" elif status == 1: source_data["status"] = "PROCESSING" elif status == 2: source_data["status"] = "UPLOADED" elif status == 3: source_data["status"] = "FAILURE" elif status == 4: source_data["status"] = "UPLOAD_FAILURE" elif status == 5: source_data["status"] = "REJECTED" if "filePath" in source_data.keys(): source_data.pop("filePath") if "savedFileName" in source_data.keys(): source_data.pop("savedFileName") if "integrationConfigId" in source_data.keys(): source_data.pop("integrationConfigId") if "metaData" in source_data.keys(): source_data.pop("metaData") if "docEntryId" in source_data.keys(): source_data.pop("docEntryId") return source_data else: # error_message = response.json()["message"] return { "message": response.status_code, }
[docs]class CogniswitchKnowledgeSourceFile(BaseTool): """Tool 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." ) """ 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. """ cs_token: str OAI_token: str apiKey: str knowledgesource_file = ( "https://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/file" ) def _run( self, file: Optional[str] = None, document_name: Optional[str] = None, document_description: Optional[str] = None, run_manager: Optional[CallbackManagerForToolRun] = None, ) -> Dict[str, Any]: """ 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. """ if not file: return { "message": "No input provided", } else: response = self.store_data( file=file, document_name=document_name, document_description=document_description, ) return response
[docs] def store_data( self, file: Optional[str], document_name: Optional[str], document_description: Optional[str], ) -> dict: """ 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. """ headers = { "apiKey": self.apiKey, "openAIToken": self.OAI_token, "platformToken": self.cs_token, } data: Dict[str, Any] if not document_name: document_name = "" if not document_description: document_description = "" if file is not None: files = {"file": open(file, "rb")} data = { "documentName": document_name, "documentDescription": document_description, } response = requests.post( self.knowledgesource_file, headers=headers, verify=False, data=data, files=files, ) if response.status_code == 200: return response.json() else: return {"message": "Bad Request"}
[docs]class CogniswitchKnowledgeSourceURL(BaseTool): """Tool 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" ) """ 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""" cs_token: str OAI_token: str apiKey: str knowledgesource_url = ( "https://api.cogniswitch.ai:8243/cs-api/0.0.1/cs/knowledgeSource/url" ) def _run( self, url: Optional[str] = None, document_name: Optional[str] = None, document_description: Optional[str] = None, run_manager: Optional[CallbackManagerForToolRun] = None, ) -> Dict[str, Any]: """ 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. """ if not url: return { "message": "No input provided", } response = self.store_data( url=url, document_name=document_name, document_description=document_description, ) return response
[docs] def store_data( self, url: Optional[str], document_name: Optional[str], document_description: Optional[str], ) -> dict: """ 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. """ headers = { "apiKey": self.apiKey, "openAIToken": self.OAI_token, "platformToken": self.cs_token, } data: Dict[str, Any] if not document_name: document_name = "" if not document_description: document_description = "" if not url: return { "message": "No input provided", } else: data = {"url": url} response = requests.post( self.knowledgesource_url, headers=headers, verify=False, data=data, ) if response.status_code == 200: return response.json() else: return {"message": "Bad Request"}