kQjUddlZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlZd Zd d d d d Zeeefed<ied d d d dZeeefed<e dGdde Zy)N)cached_property)ListOptionalDictUnion)attempt_import_or_raise)TextEmbeddingFunction)registerz!https://us-south.ml.cloud.ibm.comiii)z'ibm/granite-embedding-278m-multilingualzibm/slate-125m-english-rtrvr-v2zibm/slate-30m-english-rtrvr-v2zintfloat/multilingual-e5-largeCURRENT_MODELS)z&sentence-transformers/all-minilm-l6-v2ibm/slate-125m-english-rtrvrzibm/slate-30m-english-rtrvrz'sentence-transformers/all-minilm-l12-v2 MODELS_DIMSwatsonxceZdZUdZdZeed<dZeeed<dZ eeed<dZ eeed<dZ eeed<dZ ee ed <d eefd Zed Zed ZdZedZdeeeej0fd eeefdZedZy)WatsonxEmbeddingsus An embedding function that uses the IBM watsonx.ai Embeddings API. API Docs: https://cloud.ibm.com/apidocs/watsonx-ai#text-embeddings Supported embedding models: https://dataplatform.cloud.ibm.com/docs/content/wsj/analyze-data/fm-models-embed.html?context=wx Parameters ---------- name : str, default "ibm/slate-125m-english-rtrvr" The ID of the embedding model to use. For new tables, ``"ibm/granite-embedding-278m-multilingual"`` is recommended. api_key : str, optional IBM Cloud API key. Falls back to the ``WATSONX_API_KEY`` environment variable when not provided. project_id : str, optional watsonx.ai project ID. Explicit value takes precedence over the ``WATSONX_PROJECT_ID`` environment variable. Mutually exclusive with ``space_id`` — exactly one must be supplied. space_id : str, optional watsonx.ai deployment space ID. Explicit value takes precedence over the ``WATSONX_SPACE_ID`` environment variable. Mutually exclusive with ``project_id`` — exactly one must be supplied. url : str, optional watsonx.ai service URL. Defaults to ``"https://us-south.ml.cloud.ibm.com"``. params : dict, optional Extra parameters forwarded verbatim to ``Embeddings`` (e.g. ``{"truncate_input_tokens": 512}``). rnameNapi_key project_idspace_idurlparamsreturnc8|jr |jgSgSN)rselfs _/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/lancedb/embeddings/watsonx.py endpointszWatsonxEmbeddings.endpointsZs!XXz-2-cdgS)Nrr"r rsensitive_keysz WatsonxEmbeddings.sensitive_keys]s {r c<ttjS)aReturn the IDs of models currently available for new tables. Legacy / deprecated IDs are intentionally excluded. They remain resolvable for dimension lookups on existing tables via ``MODELS_DIMS``, but should not be used when creating new tables. )listr keysr"r r model_nameszWatsonxEmbeddings.model_namesasN'')**r c|jSr)_ndimsrs rndimszWatsonxEmbeddings.ndimsks {{r c |jtvr6td|jdttj t|jS)NzUnknown model 'z'. Available models: )rr ValueErrorr%r r&rs rr)zWatsonxEmbeddings._ndimsnsV 99K '!$))-%%).*=*=*?%@$AC 499%%r textscP|jj|dt|i|S)Nr-)_watsonx_clientembed_documentsr%)rr-argskwargss rgenerate_embeddingsz%WatsonxEmbeddings.generate_embeddingsws8 4t##33  u+   r c~td}td}|jxstjj d}|s t d|j ||jxst}|j|j}}|@|>tjj d}tjj d}|r |r t d|s |s t d t|j| }|jr|j|d <|r||d <n||d <|jdi|S)Nibm_watsonx_aiz ibm_watsonx_ai.foundation_modelsWATSONX_API_KEYzqWATSONX_API_KEY not set. Either set it in your environment or pass it as `api_key` argument to WatsonxEmbeddings.)rrWATSONX_PROJECT_IDWATSONX_SPACE_IDz4Provide either `project_id` or `space_id`, not both.zEither WATSONX_PROJECT_ID or WATSONX_SPACE_ID must be set. Pass one as an argument to WatsonxEmbeddings or set the corresponding environment variable.)model_id credentialsrrrr")r rosenvirongetr, CredentialsrDEFAULT_WATSONX_URLrrdictrr Embeddings)rr5 ibm_watsonx_ai_foundation_modelsrr:rr client_kwargss rr/z!WatsonxEmbeddings._watsonx_clients701AB+B ., ( ,,C"**..1B"CF %00//1  $}}doo*  ("2(<=Jzz~~&89H (ST T(6  #DII;O ;;&*kkM( # *4M, '(0M* %:/::K]KKr )__name__ __module__ __qualname____doc__rstr__annotations__rrrrrrrrr staticmethodr#r'r*rr)rnpndarrayfloatr3r/r"r rrr,sJ/D#.!GXc]! $J $"Hhsm"C#!FHTN!.49.++&&  T#Y *+  d5k   /L/Lr r)r; functoolsrtypingrrrrutilr baser registryr numpyrKr?r r@rHintrIrrr"r rrUs  %..*'903'*&)&* "S#X/2$'#&/2 T#s(^ )FL-FLFLr