diff --git a/guardrails/async_guard.py b/guardrails/async_guard.py index 079a6f34a..319122e55 100644 --- a/guardrails/async_guard.py +++ b/guardrails/async_guard.py @@ -97,6 +97,17 @@ async def __call( *args, **kwargs, ): + llm_api_str = "" + if llm_api: + llm_api_module_name = ( + llm_api.__module__ if hasattr(llm_api, "__module__") else "" + ) + llm_api_name = ( + llm_api.__name__ + if hasattr(llm_api, "__name__") + else type(llm_api).__name__ + ) + llm_api_str = f"{llm_api_module_name}.{llm_api_name}" if metadata is None: metadata = {} if full_schema_reask is None: @@ -111,12 +122,7 @@ async def __call( attributes=[ ("guard_id", self._guard_id), ("user_id", self._user_id), - ( - "llm_api", - llm_api.__name__ - if (llm_api and hasattr(llm_api, "__name__")) - else type(llm_api).__name__, - ), + ("llm_api", llm_api_str), ("custom_reask_prompt", self.reask_prompt is not None), ( "custom_reask_instructions", @@ -329,6 +335,17 @@ async def __parse( *args, **kwargs, ): + llm_api_str = "" + if llm_api: + llm_api_module_name = ( + llm_api.__module__ if hasattr(llm_api, "__module__") else "" + ) + llm_api_name = ( + llm_api.__name__ + if hasattr(llm_api, "__name__") + else type(llm_api).__name__ + ) + llm_api_str = f"{llm_api_module_name}.{llm_api_name}" final_num_reasks = ( num_reasks if num_reasks is not None else 0 if llm_api is None else None ) @@ -339,12 +356,7 @@ async def __parse( attributes=[ ("guard_id", self._guard_id), ("user_id", self._user_id), - ( - "llm_api", - llm_api.__name__ - if (llm_api and hasattr(llm_api, "__name__")) - else type(llm_api).__name__, - ), + ("llm_api", llm_api_str), ("custom_reask_prompt", self.reask_prompt is not None), ( "custom_reask_instructions",