Merge pull request #22 from jjmachan/fix/experiment

added git to experiment

Author: jjmachan

nbs/project/experiments.ipynb

@@ -779,7 +779,7 @@
     "# | export\n",
     "@patch\n",
     "def langfuse_experiment(\n",
-    "    self: Project, experiment_model, name_prefix: str = \"\"\n",
+    "    self: Project, experiment_model, name_prefix: str = \"\", save_to_git: bool = True, stage_all: bool = True\n",
     "):\n",
     "    \"\"\"Decorator for creating experiment functions with Langfuse integration.\n",
     "\n",
@@ -800,7 +800,7 @@
     "            return await observed_func(*args, **kwargs)\n",
     "        \n",
     "        # Now create the experiment wrapper with our already-observed function\n",
-    "        experiment_wrapper = self.experiment(experiment_model, name_prefix)(langfuse_wrapped_func)\n",
+    "        experiment_wrapper = self.experiment(experiment_model, name_prefix, save_to_git, stage_all)(langfuse_wrapped_func)\n",
     "        \n",
     "        return t.cast(ExperimentProtocol, experiment_wrapper)\n",
     "\n",

ragas_experimental/project/experiments.py

@@ -406,7 +406,13 @@ async def run_async(
 
 # %% ../../nbs/project/experiments.ipynb 32
 @patch
-def langfuse_experiment(self: Project, experiment_model, name_prefix: str = ""):
+def langfuse_experiment(
+    self: Project,
+    experiment_model,
+    name_prefix: str = "",
+    save_to_git: bool = True,
+    stage_all: bool = True,
+):
     """Decorator for creating experiment functions with Langfuse integration.
 
     Args:
@@ -428,9 +434,9 @@ async def langfuse_wrapped_func(*args, **kwargs):
             return await observed_func(*args, **kwargs)
 
         # Now create the experiment wrapper with our already-observed function
-        experiment_wrapper = self.experiment(experiment_model, name_prefix)(
-            langfuse_wrapped_func
-        )
+        experiment_wrapper = self.experiment(
+            experiment_model, name_prefix, save_to_git, stage_all
+        )(langfuse_wrapped_func)
 
         return t.cast(ExperimentProtocol, experiment_wrapper)