a bg@sddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZmZerlddlmZeeZGdddee ZdS) ) annotationsN) TYPE_CHECKINGAnyDictListOptional) Embeddings) BaseModel ConfigDict) ConnectioncseZdZUdZdZded<ded<dZded<dd fd d Zed d Z e ddddddddZ dddddZ dddddZ ZS)OracleEmbeddingszGet EmbeddingsNrconnzDict[str, Any]paramsz Optional[str]proxy)kwargsc stjfi|dS)N)super__init__)selfr __class__u/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/oracleai.pyr%szOracleEmbeddings.__init__Zforbid)extrar strNone)r dir onnx_file model_namereturnc CszN|dus"|dus"|dus"|dur*td|}|jd|||d|WnFty}z.td|t|WYd}~n d}~00dS)zLoad an ONNX model to Oracle Database. Args: conn: Oracle Connection, dir: Oracle Directory, onnx_file: ONNX file name, model_name: Name of the model. Nz Invalid inputa begin dbms_data_mining.drop_model(model_name => :model, force => true); SYS.DBMS_VECTOR.load_onnx_model(:path, :filename, :model, json('{"function" : "embedding", "embeddingOutput" : "embedding", "input": {"input": ["DATA"]}}')); end;)pathfilenamemodelAn exception occurred :: ) Exceptioncursorexecutecloseloggerinfo traceback print_exc)r rrrr$exrrrload_onnx_model3s   z OracleEmbeddings.load_onnx_modelz List[str]zList[List[float]])textsrc Csvz ddl}Wn.ty:}ztd|WYd}~n d}~00|durHdSg}zd|j_|j}|jrv|jd|jdg}t|ddD]"\}}||d } | t | q|j d } | |} |jd | t |jd |D]>} | dur| gqt | d} t | d }| |q||WStyp}z.td|t|WYd}~n d}~00dS)zCompute doc embeddings using an OracleEmbeddings. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each input text. rNzIUnable to import oracledb, please install with `pip install -U oracledb`.Fz&begin utl_http.set_proxy(:proxy); end;)r)start)Zchunk_idZ chunk_datazSYS.VECTOR_ARRAY_TzNselect t.* from dbms_vector_chain.utl_to_embeddings(:content, json(:params)) t)contentrZ embed_vectorr")oracledb ImportErrordefaultsZ fetch_lobsr r$rr% enumerateappendjsondumpsZgettypeZ newobjectrloadsr&r#r'r(r)r*)rr-r1eZ embeddingsr$chunksitextchunkZvector_array_typeinputsrowZrdataZvecr+rrrembed_documentsZsT        z OracleEmbeddings.embed_documentsz List[float])r<rcCs||gdS)zCompute query embedding using an OracleEmbeddings. Args: text: The text to embed. Returns: Embedding for the text. r)r@)rr<rrr embed_queryszOracleEmbeddings.embed_query)__name__ __module__ __qualname____doc__r __annotations__rrr Z model_config staticmethodr,r@rA __classcell__rrrrr s    &>r ) __future__rr6loggingr)typingrrrrrZlangchain_core.embeddingsrZpydanticr r r1r getLoggerrBr'r rrrr s