Ë
|
Wñúh ã ó d Z ddlmZmZmZ ddlZddlmZ ddl m
|
Z
|
ddl m Z m Z mZmZmZmZmZ ddlmZ G d d e« Zy)
|
z«Embeddings integration utilities for MySQL Connector/Python.
|
|
Provides MyEmbeddings class to generate embeddings via MySQL HeatWave
|
using ML_EMBED_TABLE and ML_EMBED_ROW.
|
é )ÚDictÚListÚOptionalN)Ú
|
Embeddings)Ú PrivateAttr)Úatomic_transactionÚ execute_sqlÚformat_value_sqlÚ source_schemaÚsql_table_from_dfÚsql_table_to_dfÚtemporary_sql_tables)ÚMySQLConnectionAbstractc ó e Zd ZU dZ e« Zeed< d dedee f fd Z
|
de e de e e fdZd e de e fd
|
Z xZS ) Ú MyEmbeddingsa
|
Embedding generator class that uses a MySQL database to compute embeddings for input text.
|
|
This class batches input text into temporary SQL tables, invokes MySQL's ML_EMBED_TABLE
|
to generate embeddings, and retrieves the results as lists of floats.
|
|
Attributes:
|
_db_connection (MySQLConnectionAbstract): MySQL connection used for all database operations.
|
schema_name (str): Name of the database schema to use.
|
options_placeholder (str): SQL-ready placeholder string for ML_EMBED_TABLE options.
|
options_params (dict): Dictionary of concrete option values to be passed as SQL parameters.
|
Ú_db_connectionÚ db_connectionÚoptionsc ó t | « || _ t |« | _ |xs i }t |« \ | _ | _ y)a
|
Initialize MyEmbeddings with a database connection and optional embedding parameters.
|
|
References:
|
https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-embed-row.html
|
A full list of supported options can be found under "options"
|
|
NOTE: The supported "options" are the intersection of the options provided in
|
https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-embed-row.html
|
https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-embed-table.html
|
|
Args:
|
db_connection: Active MySQL connector database connection.
|
options: Optional dictionary of options for embedding operations.
|
|
Raises:
|
ValueError: If the schema name is not valid
|
DatabaseError:
|
If a database connection issue occurs.
|
If an operational error occurs during execution.
|
N)ÚsuperÚ__init__r r Ú schema_namer
|
Úoptions_placeholderÚoptions_params)Úselfr r Ú __class__s úKH:\Change_password\venv_build\Lib\site-packages\mysql/ai/genai/embedding.pyr zMyEmbeddings.__init__F sC ø ô0 ÑÔØ +ÔÜ(¨ Ó7ÔØ-RÜ8HÈÓ8QÑ5Ô $Õ"5ó ÚtextsÚreturnc óú |sg S t j t t |« « |d« }t | j
|
« 5 }t | j
|
« 5 }t || j |« \ }}|j | j |f« d| d| d| j d}t ||| j ¬« t || j |« }|d j « j « st d |d D « « r t! d « |d j# « } | D
|
cg c] }
|
t% |
|
« } }
|
| cd
|
d
|
d
|
« cd
|
d
|
d
|
« S c c}
|
w # 1 sw Y nxY wd
|
d
|
d
|
« y
|
# 1 sw Y y
|
xY w) a
|
Generate embeddings for a list of input texts using the MySQL ML embedding procedure.
|
|
References:
|
https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-embed-table.html
|
|
Args:
|
texts: List of input strings to embed.
|
|
Returns:
|
List of lists of floats, with each inner list containing the embedding for a text.
|
|
Raises:
|
DatabaseError:
|
If provided options are invalid or unsupported.
|
If a database connection issue occurs.
|
If an operational error occurs during execution.
|
ValueError:
|
If one or more text entries were unable to be embedded.
|
|
Implementation notes:
|
- Creates a temporary table to pass input text to the MySQL embedding service.
|
- Adds a primary key to ensure results preserve input order.
|
- Calls ML_EMBED_TABLE and fetches the resulting embeddings.
|
- Deletes the temporary table after use to avoid polluting the database.
|
- Embedding vectors are extracted from the "embeddings" column of the result table.
|
)ÚidÚtextzCALL sys.ML_EMBED_TABLE('z .text', 'z.embeddings', ú)©ÚparamsÚ
|
embeddingsc 3 ó$ K | ] }|d u
|
y w©N© )Ú.0Úes r ú <genexpr>z/MyEmbeddings.embed_documents.<locals>.<genexpr> s è ø ò AØ T ñAùs z:Failure to generate embeddings for one or more text entry.N)ÚpdÚ DataFrameÚrangeÚlenr r r r r Úappendr r r r ÚisnullÚanyÚ
|
ValueErrorÚtolistÚlist) r r ÚdfÚcursorÚtemporary_tablesÚqualified_table_nameÚ
|
table_nameÚ embed_queryÚ df_embeddingsr' r, s r Úembed_documentsz MyEmbeddings.embed_documentsd s ñ8 ØIä \\¤¤s¨5£zÓ!2¸EÑBÓ Cô t× 2Ñ 2Ó 3ð! Ø7=Ü ×!4Ñ!4Ó 5ð! à9Iä/@Ø×(Ñ(¨"ó0Ñ ,Ð *ð × #Ñ # T×%5Ñ%5°zÐ$BÔ CðØ(Ð)ð *Ø(Ð)¨ Ø×+Ñ+Ð,Øð ð ô °D×4GÑ4GÕ Hô ,¨F°D×4DÑ4DÀjÓ QMà \Ñ*×1Ñ1Ó3×7Ñ7Ô9¼Sñ AØ#0° Ñ#>ôAô >ô !ØPóð ð
|
' |Ñ4×;Ñ;Ó=JØ+5Ö6 a$q'Ð6JÐ6à ÷C! ÷ ! ò ! ùò> 7÷?! ð ! ú÷ ! ÷ ! ñ ! ús7 ÁE1ÁCE Ä/EÅE Å E1ÅE Å E% Å!E1Å1E:r# c óä t | j « 5 }t |d| j d|g| j ¢¬« t |j « d « cddd« S # 1 sw Y yxY w)a
|
Generate an embedding for a single text string.
|
|
References:
|
https://dev.mysql.com/doc/heatwave/en/mys-hwgenai-ml-embed-row.html
|
|
Args:
|
text: The input string to embed.
|
|
Returns:
|
List of floats representing the embedding vector.
|
|
Raises:
|
DatabaseError:
|
If provided options are invalid or unsupported.
|
If a database connection issue occurs.
|
If an operational error occurs during execution.
|
|
Example:
|
>>> MyEmbeddings(db_conn).embed_query("Hello world")
|
[0.1, 0.2, ...]
|
z SELECT sys.ML_EMBED_ROW("%s", r$ r% r N)r r r r r r7 Úfetchone)r r# r9 s r r= zMyEmbeddings.embed_query¨ sn ô. × 3Ñ 3Ó 4ð .¸Ü ØØ0°×1IÑ1IÐ0JÈ!ÐLØ Ð3t× 2Ñ 2Ñ3õ ô
|
Ó)¨!Ñ,Ó-÷ .÷ .ò .ús AA&Á&A/r) )Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r Ú__annotations__r r r r ÚstrÚfloatr? r= Ú __classcell__)r s @r r r 6 su ø
ñ ñ /:«mNÐ+Ó;ð QUñ RØ 4ð RØ?Gȹ~õ Rð<B T¨#¡Yð B °4¸¸U¹ Ñ3Dó B ðH . ð .¨¨U© ÷ .r r )rE Útypingr r r Úpandasr. Úlangchain_core.embeddingsr Úpydanticr Úmysql.ai.utilsr r r
|
r r r r Úmysql.connector.abstractsr r r* r r ú<module>rP s= ðñ:÷ (Ñ 'ã å 0Ý ÷÷ ñ õ >ôO.:õ O.r
|
