o U|%i*@sdZddlZddlmZmZmZzddlmZWney'ddl mZYnwddl m Z ddl m Z mZmZddlmZGdd d eZdS) z{GenAI LLM integration utilities for MySQL Connector/Python. Provides MyLLM wrapper that issues ML_GENERATE calls via SQL. N)AnyListOptional)LLM) PrivateAttr)atomic_transaction execute_sqlformat_value_sql)MySQLConnectionAbstractc seZdZUdZeZeed<GdddZdeffdd Z dd e d e e e d e d e fd dZed efddZed e fddZZS)MyLLMa Custom Large Language Model (LLM) interface for MySQL HeatWave. This class wraps the generation functionality provided by HeatWave LLMs, exposing an interface compatible with common LLM APIs for text generation. It provides full support for generative queries and limited support for agentic queries. Attributes: _db_connection (MySQLConnectionAbstract): Underlying MySQL connector database connection. _db_connectionc@seZdZdZdZdS)z MyLLM.ConfigaL Pydantic config for the model. By default, LangChain (through Pydantic BaseModel) does not allow setting or storing undeclared attributes such as _db_connection. Setting extra = "allow" makes it possible to store extra attributes on the class instance, which is required for MyLLM. allowN)__name__ __module__ __qualname____doc__extrarrV/home/air/sos_test/back/venv/lib/python3.10/site-packages/mysql/ai/genai/generation.pyConfig@s r db_connectioncst||_dS)aS Initialize the MyLLM instance with an active MySQL database connection. Args: db_connection: A MySQL connection object used to run LLM queries. Notes: The db_connection is stored as a private attribute via object.__setattr__, which is compatible with Pydantic models. N)super__init__r )selfr __class__rrrLs zMyLLM.__init__Npromptstopkwargsreturnc Ks|}|dur ||d<t|\}}t|j%}d|d}t|||g|Rdt|dd} Wd| S1s?wY| S)ac Generate a text completion from the LLM for a given input prompt. References: https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-generate.html A full list of supported options (specified by kwargs) can be found under "options" Args: prompt: The input prompt string for the language model. stop: Optional list of stop strings to support agentic and chain-of-thought reasoning workflows. **kwargs: Additional keyword arguments providing generation options to the LLM (these are serialized to JSON and passed to the HeatWave syscall). Returns: The generated model output as a string. (The actual completion does NOT include the input prompt.) Raises: DatabaseError: If provided options are invalid or unsupported. If a database connection issue occurs. If an operational error occurs during execution. Implementation Notes: - Serializes kwargs into a SQL-compatible JSON string. - Calls the LLM stored procedure using a database cursor context. - Uses `sys.ML_GENERATE` on the server to produce the model output. - Expects the server response to be a JSON object with a 'text' key. Nstop_sequenceszSELECT sys.ML_GENERATE("%s", z);)paramsrtext)copyr rr rjsonloadsfetchone) rrrroptionsoptions_placeholderoptions_paramscursorgenerate_query llm_responserrr_call[s$    z MyLLM._callcCsddiS)z Return a dictionary of params that uniquely identify this LLM instance. Returns: dict: Dictionary of identifier parameters (should include model_name for tracing/caching). model_namemysql_heatwave_llmrrrrr_identifying_paramss zMyLLM._identifying_paramscCsdS)z Get the type name of this LLM implementation. Returns: A string identifying the LLM provider (used for logging or metrics). r/rr0rrr _llm_typeszMyLLM._llm_type)N)rrrrrr r __annotations__rrstrrrrr-propertydictr1r2 __classcell__rrrrr 0s&     2 r )rr$typingrrr#langchain_core.language_models.llmsr ImportErrorlangchain.llms.basepydanticrmysql.ai.utilsrrr mysql.connector.abstractsr r rrrrs