payi 0.1.0a72__py3-none-any.whl → 0.1.0a73__py3-none-any.whl

payi/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "payi"
-__version__ = "0.1.0-alpha.72"  # x-release-please-version
+__version__ = "0.1.0-alpha.73"  # x-release-please-version

payi/lib/AnthropicInstrumentor.py

@@ -11,7 +11,13 @@ from payi.types.ingest_units_params import Units
 from .instrument import _IsStreaming, _ProviderRequest, _PayiInstrumentor
 
 
-class AnthropicIntrumentor:
+class AnthropicInstrumentor:
+    @staticmethod
+    def is_vertex(instance: Any) -> bool:
+        from anthropic import AnthropicVertex, AsyncAnthropicVertex  # type: ignore # noqa: I001
+
+        return isinstance(instance._client, (AsyncAnthropicVertex, AnthropicVertex))
+
     @staticmethod
     def instrument(instrumentor: _PayiInstrumentor) -> None:
         try:
@@ -20,25 +26,25 @@ class AnthropicIntrumentor:
             wrap_function_wrapper(
                 "anthropic.resources.messages",
                 "Messages.create",
-                chat_wrapper(instrumentor),
+                messages_wrapper(instrumentor),
             )
 
             wrap_function_wrapper(
                 "anthropic.resources.messages",
                 "Messages.stream",
-                chat_wrapper(instrumentor),
+                messages_wrapper(instrumentor),
             )
 
             wrap_function_wrapper(
                 "anthropic.resources.messages",
                 "AsyncMessages.create",
-                achat_wrapper(instrumentor),
+                amessages_wrapper(instrumentor),
             )
 
             wrap_function_wrapper(
                 "anthropic.resources.messages",
                 "AsyncMessages.stream",
-                achat_wrapper(instrumentor),
+                amessages_wrapper(instrumentor),
             )
 
         except Exception as e:
@@ -47,7 +53,7 @@ class AnthropicIntrumentor:
 
 
 @_PayiInstrumentor.payi_wrapper
-def chat_wrapper(
+def messages_wrapper(
     instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
@@ -55,7 +61,7 @@ def chat_wrapper(
     **kwargs: Any,
 ) -> Any:
     return instrumentor.invoke_wrapper(
-        _AnthropicProviderRequest(instrumentor),
+        _AnthropicProviderRequest(instrumentor, instance),
         _IsStreaming.kwargs,
         wrapped,
         instance,
@@ -64,7 +70,7 @@ def chat_wrapper(
     )
 
 @_PayiInstrumentor.payi_awrapper
-async def achat_wrapper(
+async def amessages_wrapper(
     instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
@@ -72,7 +78,7 @@ async def achat_wrapper(
     **kwargs: Any,
 ) -> Any:
     return await instrumentor.async_invoke_wrapper(
-        _AnthropicProviderRequest(instrumentor),
+        _AnthropicProviderRequest(instrumentor, instance),
         _IsStreaming.kwargs,
         wrapped,
         instance,
@@ -81,8 +87,12 @@ async def achat_wrapper(
     )
 
 class _AnthropicProviderRequest(_ProviderRequest):
-    def __init__(self, instrumentor: _PayiInstrumentor):
-        super().__init__(instrumentor=instrumentor, category=PayiCategories.anthropic)
+    def __init__(self, instrumentor: _PayiInstrumentor, instance: Any = None) -> None:
+        self._vertex: bool = AnthropicInstrumentor.is_vertex(instance)
+        super().__init__(
+            instrumentor=instrumentor,
+            category=PayiCategories.google_vertex if self._vertex else PayiCategories.anthropic
+            )
 
     @override
     def process_chunk(self, chunk: Any) -> bool:
@@ -138,7 +148,8 @@ class _AnthropicProviderRequest(_ProviderRequest):
 
     @override
     def process_request(self, instance: Any, extra_headers: 'dict[str, str]', args: Sequence[Any], kwargs: Any) -> bool:
-        self._ingest["resource"] = kwargs.get("model", "")
+        self._ingest["resource"] = ("anthropic." if self._vertex else "") + kwargs.get("model", "")
+
         messages = kwargs.get("messages")
         if messages:
             estimated_token_count = 0 

payi/lib/OpenAIInstrumentor.py

@@ -7,7 +7,6 @@ from importlib.metadata import version
 import tiktoken  # type: ignore
 from wrapt import wrap_function_wrapper  # type: ignore
 
-from payi.types import IngestUnitsParams
 from payi.lib.helpers import PayiCategories, PayiHeaderNames
 from payi.types.ingest_units_params import Units
 
@@ -50,6 +49,18 @@ class OpenAiInstrumentor:
                 aembeddings_wrapper(instrumentor),
             )
 
+            wrap_function_wrapper(
+                "openai.resources.responses",
+                "Responses.create",
+                responses_wrapper(instrumentor),
+            )
+
+            wrap_function_wrapper(
+                "openai.resources.responses",
+                "AsyncResponses.create",
+                aresponses_wrapper(instrumentor),
+            )
+
         except Exception as e:
             logging.debug(f"Error instrumenting openai: {e}")
             return
@@ -122,10 +133,55 @@ async def achat_wrapper(
         args,
         kwargs,
     )
+    
+@_PayiInstrumentor.payi_wrapper
+def responses_wrapper(
+    instrumentor: _PayiInstrumentor,
+    wrapped: Any,
+    instance: Any,
+    *args: Any,
+    **kwargs: Any,
+) -> Any:
+    return instrumentor.invoke_wrapper(
+        _OpenAiResponsesProviderRequest(instrumentor),
+        _IsStreaming.kwargs,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    )
+
+@_PayiInstrumentor.payi_awrapper
+async def aresponses_wrapper(
+    instrumentor: _PayiInstrumentor,
+    wrapped: Any,
+    instance: Any,
+    *args: Any,
+    **kwargs: Any,
+) -> Any:
+    return await instrumentor.async_invoke_wrapper(
+        _OpenAiResponsesProviderRequest(instrumentor),
+        _IsStreaming.kwargs,
+        wrapped,
+        instance,
+        args,
+        kwargs,
+    )
 
 class _OpenAiProviderRequest(_ProviderRequest):
-    def __init__(self, instrumentor: _PayiInstrumentor):
+    chat_input_tokens_key: str = "prompt_tokens"
+    chat_output_tokens_key: str = "completion_tokens"
+    chat_input_tokens_details_key: str = "prompt_tokens_details"
+
+    responses_input_tokens_key: str = "input_tokens"
+    responses_output_tokens_key: str = "output_tokens"
+    responses_input_tokens_details_key: str = "input_tokens_details"
+
+    def __init__(self, instrumentor: _PayiInstrumentor, input_tokens_key: str, output_tokens_key: str, input_tokens_details_key: str) -> None:
         super().__init__(instrumentor=instrumentor, category=PayiCategories.openai)
+        self._input_tokens_key = input_tokens_key
+        self._output_tokens_key = output_tokens_key
+        self._input_tokens_details_key = input_tokens_details_key
 
     @override
     def process_request(self, instance: Any, extra_headers: 'dict[str, str]', args: Sequence[Any], kwargs: Any) -> bool: # type: ignore
@@ -135,16 +191,19 @@ class _OpenAiProviderRequest(_ProviderRequest):
             return True
 
         context = self._instrumentor.get_context_safe()
-        route_as_resource = extra_headers.get(PayiHeaderNames.route_as_resource) or context.get("route_as_resource")
+        price_as_category = extra_headers.get(PayiHeaderNames.price_as_category) or context.get("price_as_category")
+        price_as_resource = extra_headers.get(PayiHeaderNames.price_as_resource) or context.get("price_as_resource")
         resource_scope = extra_headers.get(PayiHeaderNames.resource_scope) or context.get("resource_scope")
 
-        if PayiHeaderNames.route_as_resource in extra_headers:
-            del extra_headers[PayiHeaderNames.route_as_resource]
+        if PayiHeaderNames.price_as_category in extra_headers:
+            del extra_headers[PayiHeaderNames.price_as_category]
+        if PayiHeaderNames.price_as_resource in extra_headers:
+            del extra_headers[PayiHeaderNames.price_as_resource]
         if PayiHeaderNames.resource_scope in extra_headers:
             del extra_headers[PayiHeaderNames.resource_scope]
             
-        if not route_as_resource:
-            logging.error("Azure OpenAI route as resource not found, not ingesting")
+        if not price_as_resource and not price_as_category:
+            logging.error("Azure OpenAI requires price as resource and/or category to be specified, not ingesting")
             return False
 
         if resource_scope:
@@ -157,8 +216,13 @@ class _OpenAiProviderRequest(_ProviderRequest):
         self._category = PayiCategories.azure_openai
 
         self._ingest["category"] = self._category
-        self._ingest["resource"] = route_as_resource
- 
+
+        if price_as_category:
+            # price as category overrides default
+            self._ingest["category"] = price_as_category
+        if price_as_resource:
+            self._ingest["resource"] = price_as_resource
+
         return True
 
     @override
@@ -193,9 +257,59 @@ class _OpenAiProviderRequest(_ProviderRequest):
 
         return True
 
+    def process_synchronous_response_worker(
+        self,
+        response: str,
+        log_prompt_and_response: bool,
+        ) -> Any:
+        response_dict = model_to_dict(response)
+
+        self.add_usage_units(response_dict.get("usage", {}))
+
+        if log_prompt_and_response:
+            self._ingest["provider_response_json"] = [json.dumps(response_dict)]
+
+        if "id" in response_dict:
+            self._ingest["provider_response_id"] = response_dict["id"]
+
+        return None
+
+    def add_usage_units(self, usage: "dict[str, Any]",) -> None:
+        units = self._ingest["units"]
+
+        input = usage[self._input_tokens_key] if self._input_tokens_key in usage else 0
+        output = usage[self._output_tokens_key] if self._output_tokens_key in usage else 0
+        input_cache = 0
+
+        prompt_tokens_details = usage.get(self._input_tokens_details_key)
+        if prompt_tokens_details:
+            input_cache = prompt_tokens_details.get("cached_tokens", 0)
+            if input_cache != 0:
+                units["text_cache_read"] = Units(input=input_cache, output=0)
+
+        input = _PayiInstrumentor.update_for_vision(input - input_cache, units, self._estimated_prompt_tokens)
+
+        units["text"] = Units(input=input, output=output)
+
+    @staticmethod
+    def has_image_and_get_texts(encoding: tiktoken.Encoding, content: Union[str, 'list[Any]'], image_type: str = "image_url", text_type:str = "text") -> 'tuple[bool, int]':
+        if isinstance(content, str):
+            return False, 0
+        elif isinstance(content, list): # type: ignore
+            has_image = any(item.get("type") == image_type for item in content)
+            if has_image is False:
+                return has_image, 0
+            
+            token_count = sum(len(encoding.encode(item.get("text", ""))) for item in content if item.get("type") == text_type)
+            return has_image, token_count
+
 class _OpenAiEmbeddingsProviderRequest(_OpenAiProviderRequest):
     def __init__(self, instrumentor: _PayiInstrumentor):
-        super().__init__(instrumentor=instrumentor)
+        super().__init__(
+            instrumentor=instrumentor,
+            input_tokens_key=_OpenAiProviderRequest.chat_input_tokens_key,
+            output_tokens_key=_OpenAiProviderRequest.chat_output_tokens_key,
+            input_tokens_details_key=_OpenAiProviderRequest.chat_input_tokens_details_key)
 
     @override
     def process_synchronous_response(
@@ -203,11 +317,16 @@ class _OpenAiEmbeddingsProviderRequest(_OpenAiProviderRequest):
         response: Any,
         log_prompt_and_response: bool,
         kwargs: Any) -> Any:
-        return process_chat_synchronous_response(response, self._ingest, log_prompt_and_response, self._estimated_prompt_tokens)
+        return self.process_synchronous_response_worker(response, log_prompt_and_response)
 
 class _OpenAiChatProviderRequest(_OpenAiProviderRequest):
     def __init__(self, instrumentor: _PayiInstrumentor):
-        super().__init__(instrumentor=instrumentor)
+        super().__init__(
+            instrumentor=instrumentor,
+            input_tokens_key=_OpenAiProviderRequest.chat_input_tokens_key,
+            output_tokens_key=_OpenAiProviderRequest.chat_output_tokens_key,
+            input_tokens_details_key=_OpenAiProviderRequest.chat_input_tokens_details_key)
+
         self._include_usage_added = False
 
     @override
@@ -223,9 +342,9 @@ class _OpenAiChatProviderRequest(_OpenAiProviderRequest):
 
         usage = model.get("usage")
         if usage:
-            add_usage_units(usage, self._ingest["units"], self._estimated_prompt_tokens)
+            self.add_usage_units(usage)
 
-            # If we aded "include_usage" in the request on behalf of the client, do not return the extra 
+            # If we added "include_usage" in the request on behalf of the client, do not return the extra 
             # packet which contains the usage to the client as they are not expecting the data
             if self._include_usage_added:
                 send_chunk_to_client = False
@@ -249,7 +368,7 @@ class _OpenAiChatProviderRequest(_OpenAiProviderRequest):
                 enc = tiktoken.get_encoding("o200k_base") # type: ignore
             
             for message in messages:
-                msg_has_image, msg_prompt_tokens = has_image_and_get_texts(enc, message.get('content', ''))
+                msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, message.get('content', ''))
                 if msg_has_image:
                     has_image = True
                     estimated_token_count += msg_prompt_tokens
@@ -276,20 +395,86 @@ class _OpenAiChatProviderRequest(_OpenAiProviderRequest):
         response: Any,
         log_prompt_and_response: bool,
         kwargs: Any) -> Any:
-        process_chat_synchronous_response(response, self._ingest, log_prompt_and_response, self._estimated_prompt_tokens)
+        return self.process_synchronous_response_worker(response, log_prompt_and_response)
+
+class _OpenAiResponsesProviderRequest(_OpenAiProviderRequest):
+    def __init__(self, instrumentor: _PayiInstrumentor):
+        super().__init__(
+            instrumentor=instrumentor,
+            input_tokens_key=_OpenAiProviderRequest.responses_input_tokens_key,
+            output_tokens_key=_OpenAiProviderRequest.responses_output_tokens_key,
+            input_tokens_details_key=_OpenAiProviderRequest.responses_input_tokens_details_key)
 
-def process_chat_synchronous_response(response: str, ingest: IngestUnitsParams, log_prompt_and_response: bool, estimated_prompt_tokens: Optional[int]) -> Any:
-    response_dict = model_to_dict(response)
+    @override
+    def process_chunk(self, chunk: Any) -> bool:
+        model = model_to_dict(chunk)
+        response: dict[str, Any] = model.get("response", {})
+
+        if "provider_response_id" not in self._ingest:
+            response_id = response.get("id", None)
+            if response_id:
+                self._ingest["provider_response_id"] = response_id
+
+        usage = response.get("usage")
+        if usage:
+            self.add_usage_units(usage)
+
+        return True
 
-    add_usage_units(response_dict.get("usage", {}), ingest["units"], estimated_prompt_tokens)
+    @override
+    def process_request(self, instance: Any, extra_headers: 'dict[str, str]', args: Sequence[Any], kwargs: Any) -> bool:
+        result = super().process_request(instance, extra_headers, args, kwargs)
+        if result is False:
+            return result
+        
+        input: ResponseInputParam  = kwargs.get("input", None) # type: ignore
+        if not input or isinstance(input, str) or not isinstance(input, list):
+            return True
+        
+        estimated_token_count = 0 
+        has_image = False
 
-    if log_prompt_and_response:
-        ingest["provider_response_json"] = [json.dumps(response_dict)]
+        try: 
+            enc = tiktoken.encoding_for_model(kwargs.get("model")) # type: ignore
+        except KeyError:
+            enc = tiktoken.get_encoding("o200k_base") # type: ignore
+        
+        # find each content..type="input_text" and count tokens
+        # input=[{
+        #     "role": "user",
+        #     "content": [
+        #         {
+        #             "type": "input_text",
+        #             "text": "what's in this image?"
+        #         },
+        #         {
+        #             "type": "input_image",
+        #             "image_url": ... 
+        #         },
+        #     ],
+        # }]
+        for item in input: # type: ignore
+            if isinstance(item, dict):
+                for key, value in item.items(): # type: ignore
+                    if key == "content":
+                        if isinstance(value, list):
+                            msg_has_image, msg_prompt_tokens = self.has_image_and_get_texts(enc, value, image_type="input_image", text_type="input_text") # type: ignore 
+                            if msg_has_image:
+                                has_image = True
+                                estimated_token_count += msg_prompt_tokens
+
+        if has_image and estimated_token_count > 0:
+            self._estimated_prompt_tokens = estimated_token_count
 
-    if "id" in response_dict:
-        ingest["provider_response_id"] = response_dict["id"]
+        return True
 
-    return None
+    @override
+    def process_synchronous_response(
+        self,
+        response: Any,
+        log_prompt_and_response: bool,
+        kwargs: Any) -> Any:
+        return self.process_synchronous_response_worker(response, log_prompt_and_response)
 
 def model_to_dict(model: Any) -> Any:
     if version("pydantic") < "2.0.0":
@@ -299,31 +484,4 @@ def model_to_dict(model: Any) -> Any:
     elif hasattr(model, "parse"):  # Raw API response
         return model_to_dict(model.parse())
     else:
-        return model
-
-
-def add_usage_units(usage: "dict[str, Any]", units: "dict[str, Units]", estimated_prompt_tokens: Optional[int]) -> None:
-    input = usage["prompt_tokens"] if "prompt_tokens" in usage else 0
-    output = usage["completion_tokens"] if "completion_tokens" in usage else 0
-    input_cache = 0
-
-    prompt_tokens_details = usage.get("prompt_tokens_details")
-    if prompt_tokens_details:
-        input_cache = prompt_tokens_details.get("cached_tokens", 0)
-        if input_cache != 0:
-            units["text_cache_read"] = Units(input=input_cache, output=0)
-
-    input = _PayiInstrumentor.update_for_vision(input - input_cache, units, estimated_prompt_tokens)
-
-    units["text"] = Units(input=input, output=output)
-
-def has_image_and_get_texts(encoding: tiktoken.Encoding, content: Union[str, 'list[Any]']) -> 'tuple[bool, int]':
-    if isinstance(content, str):
-        return False, 0
-    elif isinstance(content, list): # type: ignore
-        has_image = any(item.get("type") == "image_url" for item in content)
-        if has_image is False:
-            return has_image, 0
-        
-        token_count = sum(len(encoding.encode(item.get("text", ""))) for item in content if item.get("type") == "text")
-        return has_image, token_count
+        return model        

payi/lib/helpers.py

@@ -11,8 +11,10 @@ class PayiHeaderNames:
     use_case_id:str = "xProxy-UseCase-ID"
     use_case_name:str = "xProxy-UseCase-Name"
     use_case_version:str = "xProxy-UseCase-Version"
+    use_case_step:str = "xProxy-UseCase-Step"
     user_id:str = "xProxy-User-ID"
-    route_as_resource:str = "xProxy-RouteAs-Resource"
+    price_as_category:str = "xProxy-PriceAs-Category"
+    price_as_resource:str = "xProxy-PriceAs-Resource"
     provider_base_uri = "xProxy-Provider-BaseUri"
     resource_scope:str = "xProxy-Resource-Scope"
     api_key:str = "xProxy-Api-Key"
@@ -32,7 +34,6 @@ def create_limit_header_from_ids(limit_ids: List[str]) -> Dict[str, str]:
 
     return { PayiHeaderNames.limit_ids: ",".join(valid_ids) } if valid_ids else {}
 
-
 def create_request_header_from_tags(request_tags: List[str]) -> Dict[str, str]:
     if not isinstance(request_tags, list):  # type: ignore
         raise TypeError("request_tags must be a list")
@@ -50,7 +51,9 @@ def create_headers(
     use_case_id: Union[str, None] = None,
     use_case_name: Union[str, None] = None,
     use_case_version: Union[int, None] = None,
-    route_as_resource: Union[str, None] = None,
+    use_case_step: Union[str, None] = None,
+    price_as_category: Union[str, None] = None,
+    price_as_resource: Union[str, None] = None,
     resource_scope: Union[str, None] = None,
 ) -> Dict[str, str]:
     headers: Dict[str, str] = {}
@@ -71,8 +74,12 @@ def create_headers(
         headers.update({ PayiHeaderNames.use_case_name: use_case_name})
     if use_case_version:
         headers.update({ PayiHeaderNames.use_case_version: str(use_case_version)})
-    if route_as_resource:
-        headers.update({ PayiHeaderNames.route_as_resource: route_as_resource})
+    if use_case_step:
+        headers.update({ PayiHeaderNames.use_case_step: use_case_step})
+    if price_as_category:
+        headers.update({ PayiHeaderNames.price_as_category: price_as_category})
+    if price_as_resource:
+        headers.update({ PayiHeaderNames.price_as_resource: price_as_resource})
     if resource_scope:
         headers.update({ PayiHeaderNames.resource_scope: resource_scope })
     return headers
@@ -99,3 +106,6 @@ def payi_azure_openai_url(payi_base_url: Union[str, None] = None) -> str:
 
 def payi_aws_bedrock_url(payi_base_url: Union[str, None] = None) -> str:
     return _resolve_payi_base_url(payi_base_url=payi_base_url) + "/api/v1/proxy/aws.bedrock"
+
+# def payi_google_vertex_url(payi_base_url: Union[str, None] = None) -> str:
+#     return _resolve_payi_base_url(payi_base_url=payi_base_url) + "/api/v1/proxy/google.vertex"

payi/lib/instrument.py

@@ -98,10 +98,12 @@ class _Context(TypedDict, total=False):
     use_case_name: Optional[str]
     use_case_id: Optional[str]
     use_case_version: Optional[int]
+    use_case_step: Optional[str]
     limit_ids: Optional['list[str]']
     user_id: Optional[str]
     request_tags: Optional["list[str]"]
-    route_as_resource: Optional[str]
+    price_as_category: Optional[str]
+    price_as_resource: Optional[str]
     resource_scope: Optional[str]
 
 class _IsStreaming(Enum):
@@ -166,7 +168,7 @@ class _PayiInstrumentor:
                 global_config["proxy"] = self._proxy_default
 
             # Use default clients if not provided for global ingest instrumentation
-            if not self._payi and not self._apayi and global_config.get("proxy") is False:
+            if not self._payi and not self._apayi:
                 self._payi = Payi()
                 self._apayi = AsyncPayi()
 
@@ -211,10 +213,10 @@ class _PayiInstrumentor:
             logging.error(f"Error instrumenting OpenAI: {e}")
 
     def _instrument_anthropic(self) -> None:
-        from .AnthropicInstrumentor import AnthropicIntrumentor
+        from .AnthropicInstrumentor import AnthropicInstrumentor
 
         try:
-            AnthropicIntrumentor.instrument(self)
+            AnthropicInstrumentor.instrument(self)
 
         except Exception as e:
             logging.error(f"Error instrumenting Anthropic: {e}")
@@ -391,10 +393,12 @@ class _PayiInstrumentor:
         experience_id: Optional[str] = None,
         use_case_name: Optional[str]= None,
         use_case_id: Optional[str]= None,
-        use_case_version: Optional[int]= None,                
+        use_case_version: Optional[int]= None,
+        use_case_step: Optional[str]= None,
         user_id: Optional[str]= None,
         request_tags: Optional["list[str]"] = None,
-        route_as_resource: Optional[str] = None,
+        price_as_category: Optional[str] = None,
+        price_as_resource: Optional[str] = None,
         resource_scope: Optional[str] = None,
         ) -> None:
 
@@ -441,6 +445,7 @@ class _PayiInstrumentor:
             context["use_case_name"] = None
             context["use_case_id"] = None
             context["use_case_version"] = None
+            context["use_case_step"] = None
         else:
             if use_case_name == parent_use_case_name:
                 # Same use case name, use previous ID unless new one specified
@@ -474,10 +479,14 @@ class _PayiInstrumentor:
         else:
             context["user_id"] = user_id
 
+        if use_case_step and (context["use_case_name"] or context["use_case_id"]):
+            context["use_case_step"] = use_case_step
         if request_tags:
             context["request_tags"] = request_tags
-        if route_as_resource:
-            context["route_as_resource"] = route_as_resource
+        if price_as_category:
+            context["price_as_category"] = price_as_category
+        if price_as_resource:
+            context["price_as_resource"] = price_as_resource
         if resource_scope:
             context["resource_scope"] = resource_scope
         
@@ -497,14 +506,14 @@ class _PayiInstrumentor:
     ) -> Any:
         with self:
             self._init_current_context(
-                proxy, 
-                limit_ids, 
-                experience_name,
-                experience_id,
-                use_case_name,
-                use_case_id,
-                use_case_version,
-                user_id)
+                proxy=proxy,
+                limit_ids=limit_ids,
+                experience_name=experience_name,
+                experience_id=experience_id,
+                use_case_name=use_case_name,
+                use_case_id=use_case_id,
+                use_case_version=use_case_version,
+                user_id=user_id)
             return await func(*args, **kwargs)
 
     def _call_func(
@@ -523,14 +532,14 @@ class _PayiInstrumentor:
     ) -> Any:
         with self:
             self._init_current_context(
-                proxy, 
-                limit_ids, 
-                experience_name,
-                experience_id,
-                use_case_name,
-                use_case_id,
-                use_case_version,
-                user_id)
+                proxy=proxy,
+                limit_ids=limit_ids,
+                experience_name=experience_name,
+                experience_id=experience_id,
+                use_case_name=use_case_name,
+                use_case_id=use_case_id,
+                use_case_version=use_case_version,
+                user_id=user_id)
             return func(*args, **kwargs)
 
     def __enter__(self) -> Any:
@@ -558,13 +567,18 @@ class _PayiInstrumentor:
         args: Sequence[Any],
         kwargs: 'dict[str, Any]',
     ) -> None:
+
         limit_ids = ingest_extra_headers.pop(PayiHeaderNames.limit_ids, None)
         request_tags = ingest_extra_headers.pop(PayiHeaderNames.request_tags, None)
+
         experience_name = ingest_extra_headers.pop(PayiHeaderNames.experience_name, None)
         experience_id = ingest_extra_headers.pop(PayiHeaderNames.experience_id, None)
+
         use_case_name = ingest_extra_headers.pop(PayiHeaderNames.use_case_name, None)
         use_case_id = ingest_extra_headers.pop(PayiHeaderNames.use_case_id, None)
         use_case_version = ingest_extra_headers.pop(PayiHeaderNames.use_case_version, None)
+        use_case_step = ingest_extra_headers.pop(PayiHeaderNames.use_case_step, None)
+
         user_id = ingest_extra_headers.pop(PayiHeaderNames.user_id, None)
 
         if limit_ids:
@@ -581,6 +595,8 @@ class _PayiInstrumentor:
             request._ingest["use_case_id"] = use_case_id
         if use_case_version:
             request._ingest["use_case_version"] = int(use_case_version)
+        if use_case_step:
+            request._ingest["use_case_step"] = use_case_step
         if user_id:
             request._ingest["user_id"] = user_id
 
@@ -628,7 +644,9 @@ class _PayiInstrumentor:
             return await wrapped(*args, **kwargs)
 
         # after _udpate_headers, all metadata to add to ingest is in extra_headers, keyed by the xproxy-xxx header name
-        extra_headers = kwargs.get("extra_headers", {})
+        extra_headers: Optional[dict[str, str]] = kwargs.get("extra_headers")
+        if extra_headers is None:
+            extra_headers = {}
         self._update_extra_headers(context, extra_headers)
 
         if context.get("proxy", self._proxy_default):
@@ -729,7 +747,9 @@ class _PayiInstrumentor:
             return wrapped(*args, **kwargs)
 
         # after _udpate_headers, all metadata to add to ingest is in extra_headers, keyed by the xproxy-xxx header name
-        extra_headers = kwargs.get("extra_headers", {})
+        extra_headers: Optional[dict[str, str]] = kwargs.get("extra_headers")
+        if extra_headers is None:
+            extra_headers = {}
         self._update_extra_headers(context, extra_headers)
 
         if context.get("proxy", self._proxy_default):
@@ -836,14 +856,20 @@ class _PayiInstrumentor:
         extra_headers: "dict[str, str]",
     ) -> None:
         context_limit_ids: Optional[list[str]] = context.get("limit_ids")
+
         context_experience_name: Optional[str] = context.get("experience_name")
         context_experience_id: Optional[str] = context.get("experience_id")
+
         context_use_case_name: Optional[str] = context.get("use_case_name")
         context_use_case_id: Optional[str] = context.get("use_case_id")
         context_use_case_version: Optional[int] = context.get("use_case_version")
+        context_use_case_step: Optional[str] = context.get("use_case_step")
+
         context_user_id: Optional[str] = context.get("user_id")
         context_request_tags: Optional[list[str]] = context.get("request_tags")
-        context_route_as_resource: Optional[str] = context.get("route_as_resource")
+
+        context_price_as_category: Optional[str] = context.get("price_as_category")
+        context_price_as_resource: Optional[str] = context.get("price_as_resource")
         context_resource_scope: Optional[str] = context.get("resource_scope")
 
         # headers_limit_ids = extra_headers.get(PayiHeaderNames.limit_ids, None)
@@ -879,6 +905,7 @@ class _PayiInstrumentor:
                 extra_headers.pop(PayiHeaderNames.use_case_name, None)
                 extra_headers.pop(PayiHeaderNames.use_case_id, None)
                 extra_headers.pop(PayiHeaderNames.use_case_version, None)
+                extra_headers.pop(PayiHeaderNames.use_case_step, None)
             else:
                 # leave the value in extra_headers
                 ...
@@ -888,6 +915,8 @@ class _PayiInstrumentor:
                 extra_headers[PayiHeaderNames.use_case_id] = context_use_case_id
             if context_use_case_version is not None:
                 extra_headers[PayiHeaderNames.use_case_version] = str(context_use_case_version)
+            if context_use_case_step is not None:
+                extra_headers[PayiHeaderNames.use_case_step] = str(context_use_case_step)
 
         if PayiHeaderNames.experience_name in extra_headers:
             headers_experience_name = extra_headers.get(PayiHeaderNames.experience_name, None)
@@ -906,8 +935,11 @@ class _PayiInstrumentor:
         if PayiHeaderNames.request_tags not in extra_headers and context_request_tags:
             extra_headers[PayiHeaderNames.request_tags] = ",".join(context_request_tags)
 
-        if PayiHeaderNames.route_as_resource not in extra_headers and context_route_as_resource:
-            extra_headers[PayiHeaderNames.route_as_resource] = context_route_as_resource
+        if PayiHeaderNames.price_as_category not in extra_headers and context_price_as_category:
+            extra_headers[PayiHeaderNames.price_as_category] = context_price_as_category
+
+        if PayiHeaderNames.price_as_resource not in extra_headers and context_price_as_resource:
+            extra_headers[PayiHeaderNames.price_as_resource] = context_price_as_resource
 
         if PayiHeaderNames.resource_scope not in extra_headers and context_resource_scope:
             extra_headers[PayiHeaderNames.resource_scope] = context_resource_scope
@@ -1276,27 +1308,35 @@ def track_context(
     use_case_name: Optional[str] = None,
     use_case_id: Optional[str] = None,
     use_case_version: Optional[int] = None,
+    use_case_step: Optional[str] = None,
     user_id: Optional[str] = None,
     request_tags: Optional["list[str]"] = None,
-    route_as_resource: Optional[str] = None,
+    price_as_category: Optional[str] = None,
+    price_as_resource: Optional[str] = None,
     resource_scope: Optional[str] = None,
     proxy: Optional[bool] = None,
 ) -> _TrackContext:
     # Create a new context for tracking
     context: _Context = {}
+
+    context["proxy"] = proxy
+
     context["limit_ids"] = limit_ids
+
     context["use_case_name"] = use_case_name
     context["use_case_id"] = use_case_id
     context["use_case_version"] = use_case_version
+    context["use_case_step"] = use_case_step
+
     context["user_id"] = user_id
     context["request_tags"] = request_tags
-    context["route_as_resource"] = route_as_resource
+
+    context["price_as_category"] = price_as_category
+    context["price_as_resource"] = price_as_resource
     context["resource_scope"] = resource_scope
-    context["proxy"] = proxy
 
     return _TrackContext(context)
 
-
 @deprecated("@ingest() is deprecated. Use @track() instead")
 def ingest(
     limit_ids: Optional["list[str]"] = None,

{payi-0.1.0a72.dist-info → payi-0.1.0a73.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a72
+Version: 0.1.0a73
 Summary: The official Python library for the payi API
 Project-URL: Homepage, https://github.com/Pay-i/pay-i-python
 Project-URL: Repository, https://github.com/Pay-i/pay-i-python

{payi-0.1.0a72.dist-info → payi-0.1.0a73.dist-info}/RECORD

@@ -11,7 +11,7 @@ payi/_resource.py,sha256=j2jIkTr8OIC8sU6-05nxSaCyj4MaFlbZrwlyg4_xJos,1088
 payi/_response.py,sha256=rh9oJAvCKcPwQFm4iqH_iVrmK8bNx--YP_A2a4kN1OU,28776
 payi/_streaming.py,sha256=Z_wIyo206T6Jqh2rolFg2VXZgX24PahLmpURp0-NssU,10092
 payi/_types.py,sha256=2mbMK86K3W1aMTW7sOGQ-VND6-A2IuXKm8p4sYFztBU,6141
-payi/_version.py,sha256=hfVsfey6OwITxk6zykXfOoMOxM8ZCWk-trl1AbSPzKg,165
+payi/_version.py,sha256=rRHwLoOBgzQ6pHhvDPPLBSNIluq8Ov7y64GSnMSegaQ,165
 payi/pagination.py,sha256=k2356QGPOUSjRF2vHpwLBdF6P-2vnQzFfRIJQAHGQ7A,1258
 payi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 payi/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
@@ -25,13 +25,13 @@ payi/_utils/_transform.py,sha256=n7kskEWz6o__aoNvhFoGVyDoalNe6mJwp-g7BWkdj88,156
 payi/_utils/_typing.py,sha256=D0DbbNu8GnYQTSICnTSHDGsYXj8TcAKyhejb0XcnjtY,4602
 payi/_utils/_utils.py,sha256=ts4CiiuNpFiGB6YMdkQRh2SZvYvsl7mAF-JWHCcLDf4,12312
 payi/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
-payi/lib/AnthropicInstrumentor.py,sha256=RMl_E5rfyMT9q3bGljFA4ba3RWyURPWfgWnZe0cHV_U,7123
+payi/lib/AnthropicInstrumentor.py,sha256=o_20JNhvgRJIsGHfR0EKtWH8yLIBkafj4jKyzMDdFeI,7643
 payi/lib/BedrockInstrumentor.py,sha256=4IfOwq_MSGlvlSMxAocVu0G-Ctoir6nSL9KtpGfga8I,13969
-payi/lib/OpenAIInstrumentor.py,sha256=62aulfbZ0grdiYTDUHOiCKRHTwlrq5NHIRe6zUEMlPI,11618
+payi/lib/OpenAIInstrumentor.py,sha256=e93o253yGO-L2qiLg69BO8mc9Q2_NAcvwRFDv9qSSm8,17404
 payi/lib/Stopwatch.py,sha256=7OJlxvr2Jyb6Zr1LYCYKczRB7rDVKkIR7gc4YoleNdE,764
 payi/lib/VertexInstrumentor.py,sha256=R0s3aEW7haj4aC5t1lMeA9siOgYvxlKQShLJ-B5iiNM,11642
-payi/lib/helpers.py,sha256=2Wy_CR9PZ8rn817R5sjRJDDdJ1gItNywk0t0wtaEvmc,4058
-payi/lib/instrument.py,sha256=H4cDhMD5MKrfhjfPGKqbZBerOukUp8y3jYx853W8cEA,54072
+payi/lib/helpers.py,sha256=K1KAfWrpPT1UUGNxspLe1lHzQjP3XV5Pkh9IU4pKMok,4624
+payi/lib/instrument.py,sha256=nUUEnyDJPNMgVvT1gRgSVX1T9D8IXvBG8UzUcmfM9ws,55743
 payi/resources/__init__.py,sha256=1rtrPLWbNt8oJGOp6nwPumKLJ-ftez0B6qwLFyfcoP4,2972
 payi/resources/ingest.py,sha256=8HNHEyfgIyJNqCh0rOhO9msoc61-8IyifJ6AbxjCrDg,22612
 payi/resources/categories/__init__.py,sha256=w5gMiPdBSzJA_qfoVtFBElaoe8wGf_O63R7R1Spr6Gk,1093
@@ -141,7 +141,7 @@ payi/types/use_cases/definitions/kpi_retrieve_response.py,sha256=uQXliSvS3k-yDYw
 payi/types/use_cases/definitions/kpi_update_params.py,sha256=jbawdWAdMnsTWVH0qfQGb8W7_TXe3lq4zjSRu44d8p8,373
 payi/types/use_cases/definitions/kpi_update_response.py,sha256=zLyEoT0S8d7XHsnXZYT8tM7yDw0Aze0Mk-_Z6QeMtc8,459
 payi/types/use_cases/definitions/limit_config_create_params.py,sha256=pzQza_16N3z8cFNEKr6gPbFvuGFrwNuGxAYb--Kbo2M,449
-payi-0.1.0a72.dist-info/METADATA,sha256=6hbBjGCYyCvoUpQZnptSPqVUty2HmtPeAN8azuMtvUA,15290
-payi-0.1.0a72.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-payi-0.1.0a72.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
-payi-0.1.0a72.dist-info/RECORD,,
+payi-0.1.0a73.dist-info/METADATA,sha256=mqN0CMTKKA4ABurkrddTjGtEhu9ahrtAAwkRFjY4Nk8,15290
+payi-0.1.0a73.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+payi-0.1.0a73.dist-info/licenses/LICENSE,sha256=CQt03aM-P4a3Yg5qBg3JSLVoQS3smMyvx7tYg_6V7Gk,11334
+payi-0.1.0a73.dist-info/RECORD,,