diff --git a/pdm.lock b/pdm.lock index b68fee04..27ae559d 100644 --- a/pdm.lock +++ b/pdm.lock @@ -5,7 +5,7 @@ groups = ["default", "docs", "lint", "test"] strategy = ["cross_platform"] lock_version = "4.4.1" -content_hash = "sha256:edec106befa0b63c2236d429b7ffb3a573c8cea2dff0b46f7cd8d4efb74a4e4c" +content_hash = "sha256:4801a2394ada2b69049a40a0d522e8c53662b9262b7f8090d47e7db5d162dba0" [[package]] name = "aiohttp" @@ -3433,7 +3433,7 @@ files = [ [[package]] name = "unstract-adapters" -version = "0.21.0" +version = "0.22.0" requires_python = "<3.12,>=3.9" summary = "Unstract interface for LLMs, Embeddings and VectorDBs" dependencies = [ @@ -3461,8 +3461,8 @@ dependencies = [ "singleton-decorator~=1.0.0", ] files = [ - {file = "unstract_adapters-0.21.0-py3-none-any.whl", hash = "sha256:6c4f597602f55b80ba176a29a930755abd3494ff1c085f406536e7463902d655"}, - {file = "unstract_adapters-0.21.0.tar.gz", hash = "sha256:ebb8f80b7f26f04874bb5466fe0be8c3e1b51d44ed85c4e62289a7751c996af6"}, + {file = "unstract_adapters-0.22.0-py3-none-any.whl", hash = "sha256:0c0c4fe262ee1e3c98858ac4911c9c5accad52f92d7408e5eddef958168f786f"}, + {file = "unstract_adapters-0.22.0.tar.gz", hash = "sha256:f8ca322a1ec50e0fe257b1bb4ad6da4f6f5ca3016114437fe9119fe374e3a44b"}, ] [[package]] diff --git a/pyproject.toml b/pyproject.toml index ba14e11d..97bb7f1d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -12,7 +12,7 @@ dependencies = [ "python-magic~=0.4.27", "python-dotenv==1.0.0", # LLM Triad - "unstract-adapters~=0.21.0", + "unstract-adapters~=0.22.0", "llama-index==0.10.38", "tiktoken~=0.4.0", "transformers==4.37.0", diff --git a/src/unstract/sdk/__init__.py b/src/unstract/sdk/__init__.py index 4288fa4c..1825e5f6 100644 --- a/src/unstract/sdk/__init__.py +++ b/src/unstract/sdk/__init__.py @@ -1,4 +1,4 @@ -__version__ = "0.37.0" +__version__ = "0.38.0" def get_sdk_version(): diff --git a/src/unstract/sdk/audit.py b/src/unstract/sdk/audit.py index e9952aed..be28fefc 100644 --- a/src/unstract/sdk/audit.py +++ b/src/unstract/sdk/audit.py @@ -69,6 +69,7 @@ def push_usage_data( execution_id = kwargs.get("execution_id", "") adapter_instance_id = kwargs.get("adapter_instance_id", "") run_id = kwargs.get("run_id", "") + provider = kwargs.get("provider", "") llm_usage_reason = "" if event_type == "llm": llm_usage_reason = kwargs.get("llm_usage_reason", "") @@ -80,6 +81,7 @@ def push_usage_data( "usage_type": event_type, "llm_usage_reason": llm_usage_reason, "model_name": model_name, + "provider": provider, "embedding_tokens": token_counter.total_embedding_token_count, "prompt_tokens": token_counter.prompt_llm_token_count, "completion_tokens": token_counter.completion_llm_token_count, diff --git a/src/unstract/sdk/embedding.py b/src/unstract/sdk/embedding.py index 08b4e813..e5a71362 100644 --- a/src/unstract/sdk/embedding.py +++ b/src/unstract/sdk/embedding.py @@ -71,6 +71,7 @@ def _get_embedding(self) -> BaseEmbedding: ][Common.ADAPTER] embedding_metadata = embedding_config_data.get(Common.ADAPTER_METADATA) embedding_adapter_class = embedding_adapter(embedding_metadata) + self._usage_kwargs["provider"] = embedding_adapter_class.get_provider() return embedding_adapter_class.get_embedding_instance() except Exception as e: self._tool.stream_log( diff --git a/src/unstract/sdk/llm.py b/src/unstract/sdk/llm.py index ab83f995..f4d44f53 100644 --- a/src/unstract/sdk/llm.py +++ b/src/unstract/sdk/llm.py @@ -106,6 +106,7 @@ def _get_llm(self, adapter_instance_id: str) -> LlamaIndexLLM: ] llm_metadata = llm_config_data.get(Common.ADAPTER_METADATA) llm_adapter_class: LLMAdapter = llm_adapter(llm_metadata) + self._usage_kwargs["provider"] = llm_adapter_class.get_provider() llm_instance: LLM = llm_adapter_class.get_llm_instance() return llm_instance except Exception as e: