cellenium 0.3.2__tar.gz → 0.3.3__tar.gz

{cellenium-0.3.2 → cellenium-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: cellenium
-Version: 0.3.2
+Version: 0.3.3
 Summary: Playwright browser automation with a Google Sheets Page Object Model and CrewAI vision analysis
 Project-URL: Homepage, https://github.com/MedviJenka/cellenium-playwright
 License-File: LICENSE

{cellenium-0.3.2 → cellenium-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cellenium"
-version = "0.3.2"
+version = "0.3.3"
 description = "Playwright browser automation with a Google Sheets Page Object Model and CrewAI vision analysis"
 readme = "README.md"
 requires-python = ">=3.13"

{cellenium-0.3.2 → cellenium-0.3.3}/src/cellenium/ai/agents/vision/crew.py

@@ -1,3 +1,5 @@
+from asyncio import get_running_loop
+from concurrent.futures import ThreadPoolExecutor
 from pathlib import Path
 from cellenium.ai.config import AgentConfig
 from crewai import Agent, Crew, Task
@@ -25,8 +27,19 @@ class Vision(AgentConfig):
         return Crew(agents=self.agents, tasks=self.tasks, verbose=True)
 
 
-def run_vision_agent(image: list[str], prompt: str) -> VisionSchema:
-    return Vision().crew().kickoff({'image': image, 'prompt': prompt}).pydantic.model_dump()
+def run_vision_agent(image: list[str], prompt: str) -> dict:
+    inputs = {'image': image, 'prompt': prompt}
+
+    def kickoff() -> dict:
+        return Vision().crew().kickoff(inputs).pydantic.model_dump()
+
+    try:
+        get_running_loop()
+    except RuntimeError:
+        return kickoff()
+
+    with ThreadPoolExecutor(max_workers=1) as executor:
+        return executor.submit(kickoff).result()
 
 
 if __name__ == '__main__':

{cellenium-0.3.2 → cellenium-0.3.3}/src/cellenium/api/vision.py

@@ -1,3 +1,4 @@
+from asyncio import to_thread
 from fastapi import FastAPI, APIRouter, UploadFile, File
 from fastapi.responses import JSONResponse
 from contextlib import asynccontextmanager
@@ -6,13 +7,14 @@ from tempfile import NamedTemporaryFile
 from typing import AsyncGenerator
 from cellenium.ai.agents.vision.crew import run_vision_agent
 from cellenium.functions.logger import Logger
-from src.cellenium.settings import get_config
+from cellenium.settings import get_config
 
 
 log = Logger(name='vision-service')
 
 Config = get_config()
 
+
 @asynccontextmanager
 async def lifespan(_: FastAPI) -> AsyncGenerator:
     log.fire(f'{log.name} service started')
@@ -31,7 +33,7 @@ async def vision(prompt: str, image: UploadFile = File(...)) -> JSONResponse:
         temp_file.write(await image.read())
 
     try:
-        response = run_vision_agent(prompt=prompt, image=[str(image_path)])
+        response = await to_thread(run_vision_agent, prompt=prompt, image=[str(image_path)])
     finally:
         image_path.unlink(missing_ok=True)
 
@@ -64,4 +66,4 @@ def docs() -> JSONResponse:
 
 if __name__ == '__main__':
     import uvicorn
-    uvicorn.run(app=app, host='0.0.0.0', port=9876, use_colors=True)
+    uvicorn.run(app=app, host='0.0.0.0', port=8888, use_colors=True)

{cellenium-0.3.2 → cellenium-0.3.3}/tests/e2e/test_google_search.py

@@ -1,4 +1,5 @@
 import pytest
+from cellenium.ai.agents.vision.crew import run_vision_agent
 from cellenium.engine.page_engine import PageEngine
 from cellenium.functions.logger import Logger
 
@@ -13,5 +14,6 @@ class TestGoogleSearch:
         engine.get_web("https://www.google.com")
         engine.get_element("search").fill("cats")
         engine.get_element("button").click()
-        engine.get_screenshot()
-        log.fire('test complete')
+        sc = engine.get_screenshot()
+        response = run_vision_agent(prompt='what is displayed?', image=[sc])
+        print(response)

cellenium-0.3.3/tests/test_vision_agent.py

@@ -0,0 +1,37 @@
+import asyncio
+from types import SimpleNamespace
+
+from cellenium.ai.agents.vision import crew as vision_crew
+
+
+def test_run_vision_agent_moves_sync_kickoff_outside_active_event_loop(monkeypatch):
+    inputs = {"image": ["screenshot.png"], "prompt": "what is displayed?"}
+    expected = {"description": "search results"}
+    calls = []
+
+    class FakeCrew:
+        def kickoff(self, received_inputs: dict):
+            try:
+                asyncio.get_running_loop()
+            except RuntimeError:
+                calls.append(received_inputs)
+                return SimpleNamespace(
+                    pydantic=SimpleNamespace(model_dump=lambda: expected)
+                )
+            raise RuntimeError(
+                "Agent execution was invoked synchronously from within a running event loop."
+            )
+
+    class FakeVision:
+        def crew(self):
+            return FakeCrew()
+
+    monkeypatch.setattr(vision_crew, "Vision", FakeVision)
+
+    async def invoke_from_running_loop():
+        return vision_crew.run_vision_agent(**inputs)
+
+    response = asyncio.run(invoke_from_running_loop())
+
+    assert response == expected
+    assert calls == [inputs]

{cellenium-0.3.2 → cellenium-0.3.3}/tests/test_vision_api.py

@@ -21,7 +21,7 @@ def _load_vision_api(monkeypatch):
         "cellenium.api.vision",
         "cellenium.functions.logger",
         "src.cellenium.api.vision",
-        "src.cellenium.settings",
+        "cellenium.settings",
     ):
         sys.modules.pop(module_name, None)