7|hddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZmZerddlmZej$eZ Gddee Zy) ) annotationsN) TYPE_CHECKINGAnyDictListOptional) Embeddings) BaseModel ConfigDict) ConnectionceZdZUdZ dZded< ded< dZded<dfd Zed Z e dd Z dd Z ddZ xZS)OracleEmbeddingszGet EmbeddingsNrconnzDict[str, Any]paramsz Optional[str]proxyc $t|di|y)N)super__init__)selfkwargs __class__s f/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/oracleai.pyrzOracleEmbeddings.__init__%s "6"forbid)extrac2 |||| td|j}|jd||||jy#t$rB}tj d|t jjd}~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)rdir onnx_file model_namer#exs rload_onnx_modelz OracleEmbeddings.load_onnx_model3s |s{i.?:CU00[[]F NN"    LLN  KK3B48 9    ! LLN   sAA B=BBc ddl}|yg} d|j_|jj }|j r|jd|j g}t|dD].\}}||d } |jtj| 0|jjd } | j|} |jd | tj|j |D]W} | |jgtj| d} tj| d }|j|Y|j!|S#t$r}td|d}~wwxYw#t"$rB}t$j'd|t)j*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))rtextsr6e embeddingsr#chunksitextchunkvector_array_typeinputsrowrdatavecr-s rembed_documentsz OracleEmbeddings.embed_documentsZs   =(* ( +0H   (YY%%'Fzz%:;C%%c* + LLN [ -  \  KK3B48 9    ! LLN   s/E+EF+ F4 FF G=GGc,|j|gdS)zCompute query embedding using an OracleEmbeddings. Args: text: The text to embed. Returns: Embedding for the text. r)rM)rrFs r embed_queryzOracleEmbeddings.embed_querys##TF+A..r)rr) rr r*strr+rPr,rPreturnNone)rAz List[str]rQzList[List[float]])rFrPrQz List[float])__name__ __module__ __qualname____doc__r__annotations__rrr model_config staticmethodr.rMrO __classcell__)rs@rrrsD# E=#L$$"$/2$@C$ $$L<|/rr) __future__rr<loggingr(typingrrrrrlangchain_core.embeddingsr pydanticr r r6r getLoggerrSr&rrrrrasP# ;;0*#   8 $D/y*D/Rr