Ng FddlmZmZmZmZddlmZGddeZdS))AnyDictListOptional) GraphStorec >eZdZdZdeddfdZedefdZedee effdZ de fdZ deddfd Z dd e ee efddfd Zdee eee efffd Zd Zde dee effdZifde de de dedee f dZdS) TigerGraphaTigerGraph wrapper for graph operations. *Security note*: Make sure that the database connection uses credentials that are narrowly-scoped to only include necessary permissions. Failure to do so may result in data corruption or loss, since the calling code may attempt commands that would result in deletion, mutation of data if appropriately prompted or reading sensitive data if such data is present in the database. The best way to guard against such negative outcomes is to (as appropriate) limit the permissions granted to the credentials used with this tool. See https://python.langchain.com/docs/security for more information. connreturnNcX|||dS)z/Create a new TigerGraph graph wrapper instance.N)set_connection set_schema)selfr s g/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/graphs/tigergraph_graph.py__init__zTigerGraph.__init__s, D!!! c|jSN)_connrs rr zTigerGraph.conns zrc|jSr)_schemars rschemazTigerGraph.schemas |rc|jrt|jS|t|jSr)rstrrrs r get_schemazTigerGraph.get_schema"s< < %t|$$ $ OO   t|$$ $rc ddlm}n#t$rtdwxYwt||sd}t ||jjd}t|||_| dS)Nr)TigerGraphConnectionz`Could not import pyTigerGraph python package. Please install it with `pip install pyTigerGraph`.z9**conn** parameter must inherit from TigerGraphConnectionzs**conn** parameter does not have nlqs_host parameter defined. Define hostname of NLQS service.) pyTigerGraphr ImportError isinstance TypeErrorai nlqs_hostConnectionErrorrr)rr rmsgs rr zTigerGraph.set_connection)s  9 9 9 9 9 9 9   E   $ 455 !MCC.. 7  $9C!#&& &+/  s #rc@||n||_dS)zq Set the schema of the TigerGraph Database. Auto-generates Schema if **schema** is None. N)generate_schemar)rrs rrzTigerGraph.set_schema>s$ 28t++---V rc8|jdS)a  Generates the schema of the TigerGraph Database and returns it User can specify a **sample_ratio** (0 to 1) to determine the ratio of documents/edges used (in relation to the Collection size) to render each Collection Schema. T)force)r getSchemars rr(zTigerGraph.generate_schemaEsz##$#///rc.|dSr)r(rs rrefresh_schemazTigerGraph.refresh_schemaPs rquerycD|jj|}|S)zQuery the TigerGraph database.)rr#r.)rr.answers rr.zTigerGraph.querySs$$U++ rfunction_header description docstring param_typescF|jj||||S)zZ Wrapper function to register a custom GSQL query to the TigerGraph NLQS. )rr#registerCustomQuery)rr1r2r3r4s rregister_queryzTigerGraph.register_queryXs)z}00 [)[   rr)__name__ __module__ __qualname____doc__rrpropertyr rrrrr rrrr(r-r.dictr7rrr r s  ST cXS#XX%C%%%%34*LL$sCx.!9LTLLLL 0 c4S#X'' ( 0 0 0 034S>           c       rr N)typingrrrr&langchain_community.graphs.graph_storerr r>rrrAsy,,,,,,,,,,,,======^ ^ ^ ^ ^ ^ ^ ^ ^ ^ r