U h@sddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZerhddlmZeeZGdddee ZdS) ) annotationsN) TYPE_CHECKINGAnyDictListOptional) Embeddings) BaseModel) ConnectioncseZdZUdZded<ded<dZded<dd fd d ZGd d d ZeddddddddZ dddddZ dddddZ Z S)OracleEmbeddingszGet EmbeddingsrconnzDict[str, Any]paramsNz Optional[str]proxy)kwargsc stjf|dS)N)super__init__)selfr __class__K/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/oracleai.pyr%szOracleEmbeddings.__init__c@seZdZdZdS)zOracleEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrarrrrConfig(srr strNone)r dir onnx_file model_namereturnc CszN|dks"|dks"|dks"|dkr*td|}|jd|||d|WnDtk r}z&td|t|W5d}~XYnXdS)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 rrr r'exrrrload_onnx_model2s   z OracleEmbeddings.load_onnx_modelz List[str]zList[List[float]])textsr!c Csrz ddl}Wn,tk r8}ztd|W5d}~XYnX|dkrFdSg}zd|j_|j}|jrt|jd|jdg}t|ddD]"\}}||d } | t | q|j d } | |} |jd | t |jd |D]>} | dkr| gqt | d} t | d }| |q||WStk rl}z&td|t|W5d}~XYnXdS)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)contentr Z embed_vectorr%)oracledb ImportErrordefaultsZ fetch_lobsr r'rr( enumerateappendjsondumpsZgettypeZ newobjectr loadsr)r&r*r+r,r-)rr0r4eZ embeddingsr'chunksitextchunkZvector_array_typeinputsrowZrdataZvecr.rrrembed_documentsYsV        z OracleEmbeddings.embed_documentsz List[float])r?r!cCs||gdS)zCompute query embedding using an OracleEmbeddings. Args: text: The text to embed. Returns: Embedding for the text. r)rC)rr?rrr embed_queryszOracleEmbeddings.embed_query) rrr__doc____annotations__rrr staticmethodr/rCrD __classcell__rrrrr s   &>r ) __future__rr9loggingr,typingrrrrrZlangchain_core.embeddingsrZlangchain_core.pydantic_v1r r4r getLoggerrr*r rrrr s