snowglobe 0.4.11__tar.gz → 0.4.12__tar.gz

{snowglobe-0.4.11/src/snowglobe.egg-info → snowglobe-0.4.12}/PKG-INFO

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

{snowglobe-0.4.11 → snowglobe-0.4.12}/pyproject.toml

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

{snowglobe-0.4.11 → snowglobe-0.4.12}/src/snowglobe/client/src/app.py

@@ -157,7 +157,11 @@ async def process_application_heartbeat(app_id):
         )
         async def run_completion_fn(completion_request: CompletionRequest):
             if asyncio.iscoroutinefunction(completion_fn):
-                response = await completion_fn(completion_request)
+                try:
+                    asyncio.get_running_loop()
+                    response = await completion_fn(completion_request)
+                except RuntimeError:
+                    response = asyncio.run(completion_fn(completion_request))
             else:
                 response = completion_fn(completion_request)
             return response
@@ -250,7 +254,11 @@ async def process_risk_evaluation(test, risk_name, simulation_name, agent_name):
     )
     async def run_risk_evaluation_fn(risk_evaluation_request: RiskEvaluationRequest):
         if asyncio.iscoroutinefunction(risks[risk_name]):
-            risk_evaluation = await risks[risk_name](risk_evaluation_request)
+            try:
+                asyncio.get_running_loop()
+                risk_evaluation = await risks[risk_name](risk_evaluation_request)
+            except RuntimeError:
+                risk_evaluation = asyncio.run(risks[risk_name](risk_evaluation_request))
         else:
             risk_evaluation = risks[risk_name](risk_evaluation_request)
         return risk_evaluation
@@ -312,7 +320,11 @@ async def process_test(test, completion_fn, app_id, simulation_name):
     )
     async def run_completion_fn(completion_request: CompletionRequest):
         if asyncio.iscoroutinefunction(completion_fn):
-            completionOutput = await completion_fn(completion_request)
+            try: 
+                asyncio.get_running_loop()
+                completionOutput = await completion_fn(completion_request)
+            except RuntimeError:
+                completionOutput = asyncio.run(completion_fn(completion_request))
         else:
             completionOutput = completion_fn(completion_request)
         return completionOutput

{snowglobe-0.4.11 → snowglobe-0.4.12}/src/snowglobe/client/src/cli.py

@@ -401,7 +401,11 @@ def test_agent_wrapper(filename: str, app_id: str, app_name: str) -> Tuple[bool,
         )
         async def run_process_scenario(completion_request: CompletionRequest):
             if asyncio.iscoroutinefunction(process_scenario):
-                response = asyncio.run(process_scenario(completion_request))
+                try:
+                    asyncio.get_running_loop()
+                    response = await process_scenario(completion_request)
+                except RuntimeError:
+                    response = asyncio.run(process_scenario(completion_request))
             else:
                 response = process_scenario(completion_request)
             return response

{snowglobe-0.4.11 → snowglobe-0.4.12/src/snowglobe.egg-info}/PKG-INFO

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