From c0e557ada9747ba328368c63a181fd698dcb03d5 Mon Sep 17 00:00:00 2001
From: vasilije <vas.markovic@gmail.com>
Date: Sat, 12 Jul 2025 11:53:05 +0200
Subject: [PATCH] added basic BAML test

---
 .github/workflows/basic_tests.yml | 33 +++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/.github/workflows/basic_tests.yml b/.github/workflows/basic_tests.yml
index 13ea829da..4bc0ae821 100644
--- a/.github/workflows/basic_tests.yml
+++ b/.github/workflows/basic_tests.yml
@@ -153,6 +153,39 @@ jobs:
       - name: Run Simple Examples
         run: poetry run python ./examples/python/simple_example.py
 
+  simple-examples-baml:
+    name: Run Simple Examples BAML
+    runs-on: ubuntu-22.04
+    env:
+      GRAPHISTRY_USERNAME: ${{ secrets.GRAPHISTRY_USERNAME }}
+      GRAPHISTRY_PASSWORD: ${{ secrets.GRAPHISTRY_PASSWORD }}
+
+      LLM_PROVIDER: openai
+      LLM_MODEL: ${{ secrets.LLM_MODEL }}
+      LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
+      LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
+      LLM_API_VERSION: ${{ secrets.LLM_API_VERSION }}
+
+      EMBEDDING_PROVIDER: openai
+      EMBEDDING_MODEL: ${{ secrets.EMBEDDING_MODEL }}
+      EMBEDDING_ENDPOINT: ${{ secrets.EMBEDDING_ENDPOINT }}
+      EMBEDDING_API_KEY: ${{ secrets.EMBEDDING_API_KEY }}
+      EMBEDDING_API_VERSION: ${{ secrets.EMBEDDING_API_VERSION }}
+      STRUCTURED_OUTPUT_FRAMEWORK: "BAML"
+    steps:
+      - name: Check out repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
+      - name: Cognee Setup
+        uses: ./.github/actions/cognee_setup
+        with:
+          python-version: ${{ inputs.python-version }}
+
+      - name: Run Simple Examples
+        run: poetry run python ./examples/python/simple_example.py
+
   graph-tests:
     name: Run Basic Graph Tests
     runs-on: ubuntu-22.04