test: beeai compiler test

Signed-off-by: Nick Mitchell <[email protected]>

Author: starpit

.github/workflows/tauri-cli.yml

@@ -36,7 +36,17 @@ jobs:
       - name: Setup xvfb for screen 0
         run: Xvfb :1 -screen 0 1600x1200x24 &
 
-      - name: Run production build
+      - name: Test beeai compiler
+        env:
+          DISPLAY: :1
+        run: |
+          PATH=./src-tauri/target/release/:$PATH
+
+          for i in ./demos/beeai/*.py
+          do pdl compile beeai $i -g -o /tmp/z.json && jq .description /tmp/z.json
+          done
+          
+      - name: Test pdl run against production build
         env:
           DISPLAY: :1
         run: |
@@ -64,4 +74,4 @@ jobs:
           done
 
       - name: Tear down xvfb
-        run: killall Xvfb  
+        run: killall Xvfb || true

pdl-live-react/demos/beeai/test1.py

@@ -0,0 +1,64 @@
+import asyncio
+
+from beeai_framework.backend.chat import ChatModel
+from beeai_framework.tools.search.wikipedia import WikipediaTool
+from beeai_framework.tools.weather.openmeteo import OpenMeteoTool
+from beeai_framework.workflows.agent import AgentWorkflow, AgentWorkflowInput
+
+
+async def main() -> None:
+    llm = ChatModel.from_name("ollama:granite3.2:2b")
+    workflow = AgentWorkflow(name="Smart assistant")
+
+    workflow.add_agent(
+        name="Researcher",
+        role="A diligent researcher.",
+        instructions="You look up and provide information about a specific topic.",
+        tools=[WikipediaTool()],
+        llm=llm,
+    )
+
+    workflow.add_agent(
+        name="WeatherForecaster",
+        role="A weather reporter.",
+        instructions="You provide detailed weather reports.",
+        tools=[OpenMeteoTool()],
+        llm=llm,
+    )
+
+    workflow.add_agent(
+        name="DataSynthesizer",
+        role="A meticulous and creative data synthesizer",
+        instructions="You can combine disparate information into a final coherent summary.",
+        llm=llm,
+    )
+
+    location = "Saint-Tropez"
+
+    response = await workflow.run(
+        inputs=[
+            AgentWorkflowInput(
+                prompt=f"Provide a short history of {location}.",
+            ),
+            AgentWorkflowInput(
+                prompt=f"Provide a comprehensive weather summary for {location} today.",
+                expected_output="Essential weather details such as chance of rain, temperature and wind. Only report information that is available.",
+            ),
+            AgentWorkflowInput(
+                prompt=f"Summarize the historical and weather data for {location}.",
+                expected_output=f"A paragraph that describes the history of {location}, followed by the current weather conditions.",
+            ),
+        ]
+    ).on(
+        "success",
+        lambda data, event: print(
+            f"\n-> Step '{data.step}' has been completed with the following outcome.\n\n{data.state.final_answer}"
+        ),
+    )
+
+    print("==== Final Answer ====")
+    print(response.result.final_answer)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

pdl-live-react/package.json

@@ -14,9 +14,9 @@
     "tauri": "tauri",
     "test:quality": "concurrently -n 'lint,types,formatting' 'npm run lint' 'tsc --build --noEmit' \"prettier --check 'tests/**/*.ts'  'src/**/*.{ts,tsx,css}'\"",
     "test:ui": "playwright install --with-deps && playwright test",
-    "test:bee": "./src-tauri/target/debug/pdl compile beeai demos/beeai.py -g --output - | jq", 
+    "test:bee": "until [ -f ./src-tauri/target/debug/pdl ]; do sleep 1; done; for i in ./demos/beeai/*.py; do ./src-tauri/target/debug/pdl compile beeai $i -g --output - | jq; done", 
     "types": "(cd .. && python -m src.pdl.pdl --schema > src/pdl/pdl-schema.json) && json2ts ../src/pdl/pdl-schema.json src/pdl_ast.d.ts --unreachableDefinitions && npm run format",
-    "test": "concurrently -n 'quality,playwright,bee' 'npm run test:quality' 'npm run test:ui' 'npm run test:bee'",
+    "test": "concurrently -n 'quality,playwright' 'npm run test:quality' 'npm run test:ui'",
     "pdl": "./src-tauri/target/debug/pdl",
     "view": "npm run pdl view",
     "start": "npm run tauri dev"

pdl-live-react/src-tauri/src/interpreter/pip.rs

@@ -25,9 +25,13 @@ pub async fn pip_install_if_needed(
         } else {
             "python3"
         };
-        cmd!(python, "-mvenv", &venv_path).run()?;
+        cmd!(python, "-mvenv", &venv_path)
+            .stdout_to_stderr()
+            .run()?;
 
-        cmd!(bin_path.join("pip"), "install", "-r", &requirements_path,).run()?;
+        cmd!(bin_path.join("pip"), "install", "-r", &requirements_path,)
+            .stdout_to_stderr()
+            .run()?;
 
         let cached_requirements_path = venv_path.join("requirements.txt");
         copy(requirements_path, cached_requirements_path)?;

pdl-live-react/src-tauri/src/interpreter/pull.rs

@@ -42,7 +42,10 @@ fn ollama_exists(model: &str) -> bool {
 /// The Ollama implementation of a single model pull
 fn ollama_pull_if_needed(model: &str) -> Result<(), LoadError> {
     if !ollama_exists(model) {
-        cmd!("ollama", "pull", model).run().map_err(LoadError::IO)?;
+        cmd!("ollama", "pull", model)
+            .stdout_to_stderr()
+            .run()
+            .map_err(LoadError::IO)?;
     }
     Ok(())
 }

pyproject.toml

@@ -78,3 +78,6 @@ where = ["src"]
 
 [tool.setuptools.package-data]
 pdl = ["pdl-schema.json"]
+
+[tool.pyright]
+include = ["src", "tests", "examples", "docs"]