snowglobe 0.4.5__tar.gz → 0.4.6__tar.gz

{snowglobe-0.4.5/src/snowglobe.egg-info → snowglobe-0.4.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe
-Version: 0.4.5
+Version: 0.4.6
 Summary: client server for usage with snowglobe experiments
 Author-email: Guardrails AI <contact@guardrailsai.com>
 License: MIT License

{snowglobe-0.4.5 → snowglobe-0.4.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "snowglobe"
-version = "0.4.5"
+version = "0.4.6"
 authors = [
     {name = "Guardrails AI", email = "contact@guardrailsai.com"}
 ]

{snowglobe-0.4.5 → snowglobe-0.4.6}/src/snowglobe/client/src/app.py

@@ -151,18 +151,18 @@ async def process_application_heartbeat(app_id):
                 f"Completion function for application {app_id} did not return a valid response. Expected a string, got {type(response.response)}"
             )
             connection_test_payload["status"] = "failed"
-            connection_test_payload["error"] = (
-                "Completion function did not return a valid response. Expected a string, got {type(response.response)}"
-            )
+            connection_test_payload[
+                "error"
+            ] = "Completion function did not return a valid response. Expected a string, got {type(response.response)}"
 
         if response.response == "This is a string response from your application":
             LOGGER.error(
                 f"Completion function for application {app_id} returned a default response. This indicates the application is not properly connected."
             )
             connection_test_payload["status"] = "failed"
-            connection_test_payload["error"] = (
-                "Completion function returned a default response. "
-            )
+            connection_test_payload[
+                "error"
+            ] = "Completion function returned a default response. "
 
         if connection_test_payload.get("status") != "failed":
             connection_test_payload["response"] = response.response
@@ -693,17 +693,8 @@ async def lifespan(app: FastAPI):
                             )
                             continue
 
-                        # Wrap process_scenario to match the expected completion_fn interface
-                        def make_completion_fn(process_fn):
-                            def completion_fn(request):
-                                return process_fn(request)
-
-                            return completion_fn
-
                         apps[app_id] = {
-                            "completion_fn": make_completion_fn(
-                                agent_module.process_scenario
-                            ),
+                            "completion_fn": agent_module.process_scenario,
                             "name": app_name,
                         }
 

{snowglobe-0.4.5 → snowglobe-0.4.6}/src/snowglobe/client/src/cli_utils.py

@@ -270,7 +270,7 @@ def graceful_shutdown(_sig_num, _frame):
 
     if cli_state.verbose:
         console.print(f"signal received: {_sig_num}, frame: {_frame}")
-    
+
     # Show session summary and exit cleanly
     console.print("\n[bold blue]🛑 Shutting down...[/bold blue]")
     show_session_summary()

{snowglobe-0.4.5 → snowglobe-0.4.6/src/snowglobe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowglobe
-Version: 0.4.5
+Version: 0.4.6
 Summary: client server for usage with snowglobe experiments
 Author-email: Guardrails AI <contact@guardrailsai.com>
 License: MIT License

{snowglobe-0.4.5 → snowglobe-0.4.6}/tests/test_cli.py

@@ -531,12 +531,14 @@ class TestCli(unittest.TestCase):
             # Test Case 19: Valid agent with default template response
             template_file = os.path.join(tmpdir, "template.py")
             with open(template_file, "w") as f:
-                f.write("""
+                f.write(
+                    """
 from snowglobe.client import CompletionRequest, CompletionFunctionOutputs
 
 def process_scenario(request):
     return CompletionFunctionOutputs(response="Your response here")
-""")
+"""
+                )
 
             with mock.patch("os.getcwd", return_value=tmpdir):
                 result, message = cli.test_agent_wrapper(