NgddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZmZerddlmZejeZ Gddee ZdS) ) annotationsN) TYPE_CHECKINGAnyDictListOptional) Embeddings) BaseModel ConfigDict) ConnectionceZdZUdZ dZded< ded< dZded<dfd Zed Z e ddZ ddZ ddZ xZS) OracleEmbeddingszGet EmbeddingsNrconnzDict[str, Any]paramsz Optional[str]proxykwargsc :tjdi|dS)N)super__init__)selfr __class__s c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/oracleai.pyrzOracleEmbeddings.__init__%s&""6"""""forbid)extrar dirstr onnx_file model_namereturnNonecf ||||td|}|d||||dS#t$rJ}td|t j|d}~wwxYw)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)rrrr r)exs rload_onnx_modelz OracleEmbeddings.load_onnx_model3s |s{i.?:CU000[[]]F NN"      LLNNNNN    KK8B88 9 9 9   ! ! ! LLNNN   sAA B0&AB++B0texts List[str]List[List[float]]c ddl}n"#t$r}td|d}~wwxYw|dSg} d|j_|j}|jr|d|jg}t|dD]1\}}||d } | tj | 2|j d } | |} |d | tj |j |D]c} | | gtj| d} tj| d }| |d||S#t"$rJ}t$d|t)j|d}~wwxYw)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)chunk_id chunk_datazSYS.VECTOR_ARRAY_TzNselect t.* from dbms_vector_chain.utl_to_embeddings(:content, json(:params)) t)contentr embed_vectorr')oracledb ImportErrordefaults fetch_lobsrr)rr* enumerateappendjsondumpsgettype newobjectrloadsr+r(r,r-r.r/)rr2r<e embeddingsr)chunksitextchunkvector_array_typeinputsrowrdatavecr0s rembed_documentsz OracleEmbeddings.embed_documentsZs5  OOOO   -   =4(* ( +0H  (Y%%''Fz %:;;C%%c**** LLNNN     KK8B88 9 9 9   ! ! ! LLNNN   s) &!&E&F G+!AG&&G+rK List[float]c:||gdS)zCompute query embedding using an OracleEmbeddings. Args: text: The text to embed. Returns: Embedding for the text. r)rR)rrKs r embed_queryzOracleEmbeddings.embed_querys##TF++A..r)rr) rr rrrrr rr!r")r2r3r!r4)rKrr!rS)__name__ __module__ __qualname____doc__r__annotations__rrr model_config staticmethodr1rRrU __classcell__)rs@rrrsDE######:L$$$\$L<<<<|////////rr) __future__rrBloggingr.typingrrrrrlangchain_core.embeddingsr pydanticr r r<r getLoggerrVr,rrrrrds#""""" ;;;;;;;;;;;;;;000000********$######  8 $ $D/D/D/D/D/y*D/D/D/Rr