Track hyperparameters for each LLM function call, too

BlackHC · Nov 11, 2023 · 044d3fe · 044d3fe
1 parent a6a61bd
commit 044d3fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/llm_strategy/llm_function.py b/llm_strategy/llm_function.py
@@ -874,7 +874,7 @@ def apply_decorator(f: F_types, decorator) -> F_types:
         if not is_not_implemented(f):
             raise ValueError("The function must not be implemented.")
 
-        specific_llm_function = functools.wraps(f)(decorator(f))
+        specific_llm_function = track_hyperparameters(functools.wraps(f)(decorator(f)))
 
     return typing.cast(F_types, specific_llm_function)
 

diff --git a/llm_strategy/llm_strategy.py b/llm_strategy/llm_strategy.py
@@ -37,7 +37,7 @@ def can_wrap_function_in_llm(f: typing.Callable[P, T]) -> bool:
 
 def llm_strategy(llm: BaseLLM) -> typing.Callable[[T], T]:  # noqa: C901
     """
-    A strategy that implements what ever it decorates (or is called on) using the LLM.
+    A strategy that implements whatever it decorates (or is called on) using the LLM.
     """
 
     @typing.no_type_check