tj vdZddlZddlmZmZmZmZddlmZm Z ej e Z GddeZ dS)z4Utils for interacting with the Semantic Scholar API.N)AnyDictListOptional) BaseModelmodel_validatorceZdZUdZeed<dZeed<dZeed<dZ eed<d Z e eed <gd Z e eed <ed ededefdZdedefdZdS)SemanticScholarAPIWrapperaWrapper around semanticscholar.org API. https://github.com/danielnsilva/semanticscholar You should have this library installed. `pip install semanticscholar` Semantic Scholar API can conduct searches and fetch document metadata like title, abstract, authors, etc. Attributes: top_k_results: number of the top-scored document used for the Semantic Scholar tool load_max_docs: a limit to the number of loaded documents Example: .. code-block:: python from langchain_community.utilities.semanticscholar import SemanticScholarAPIWrapper ss = SemanticScholarAPIWrapper( top_k_results = 3, load_max_docs = 3 ) ss.run("biases in large language models") semanticscholar_search top_k_resultsi,S2_MAX_QUERY_LENGTHd load_max_docsidoc_content_chars_max) titleabstractvenueyearpaperId citationCount openAccessPdfauthors externalIdsreturned_fieldsbefore)modevaluesreturncx ddlm}|}|j|d<n#t$rtdwxYw|S)z7Validate that the python package exists in environment.r)SemanticScholarr zfCould not import Semanticscholar python package. Please install it with `pip install semanticscholar`.)semanticscholarr! search_paper ImportError)clsrr!schs k/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/utilities/semanticscholar.pyvalidate_environmentz.SemanticScholarAPIWrapper.validate_environment;sq  7 7 7 7 7 7!/##C/2/?F+ , ,   H    s7queryc|||j|j}g}|d|jD]}ddt |dgD}|dt |dddt |d dd |d t |d dd |r"d|d|jSdS)zRun the Semantic Scholar API.)limitfieldsNz, c3&K|] }|dV dS)nameN).0authors r' z0SemanticScholarAPIWrapper.run..Rs7  #)v      rzPublished year: rz Title: rz Authors: z Abstract: r z zNo results found.)r rrr joingetattrappendr)selfr)results documentsitemrs r'runzSemanticScholarAPIWrapper.runKsD-- +D4H.   0d001  Dii  -4T9b-I-I   G   A74#>#>AA!$66AA#AA%T:t<<AAA      ';;y))*FD,F*FG G&&r3N)__name__ __module__ __qualname____doc__r__annotations__r intrrrrrrstrr classmethodrr(r<r/r3r'r r s2 M3""""M3+/8C=/// " " "OT#Y   _(### $ 3   [$# ''''''''r3r )r@loggingtypingrrrrpydanticrr getLoggerr=loggerr r/r3r'rJs::,,,,,,,,,,,,////////  8 $ $T'T'T'T'T' T'T'T'T'T'r3