add llm tests

Author: JR-1991

.github/workflows/test.yml

@@ -26,6 +26,11 @@ jobs:
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python-version }}
+      - name: Pull Ollama model
+        run: |
+          curl http://localhost:11434/api/pull -d '{
+            "model": "gemma2:2b"
+          }'
       - name: Install dependencies
         run: |
           python3 -m pip install --upgrade pip
@@ -43,6 +48,8 @@ jobs:
           NEO4J_PASSWORD: your_password
           NEO4J_HOST: localhost
           NEO4J_PORT: 7687
+          OLLAMA_HOST: "http://localhost:11434/v1"
+          OLLAMA_MODEL: "gemma2:2b"
         run: |
           python3 -m poetry run pytest
 
@@ -54,3 +61,7 @@ jobs:
           - 7687:7687
         env:
           NEO4J_AUTH: neo4j/your_password
+      ollama:
+        image: ollama/ollama:latest
+        ports:
+          - 11434:11434

tests/integration/test_llm.py

@@ -0,0 +1,27 @@
+import os
+from mdmodels import DataModel, llm
+
+
+class TestLLM:
+    def test_llm(self):
+        # Arrange
+        OLLAMA_HOST = os.getenv("OLLAMA_HOST")
+        OLLAMA_MODEL = os.getenv("OLLAMA_MODEL")
+
+        assert OLLAMA_HOST, "OLLAMA_HOST is not set"
+        assert OLLAMA_MODEL, "OLLAMA_MODEL is not set"
+
+        dm = DataModel.from_markdown("./tests/fixtures/model_graph.md")
+        input_text = "My name is John Doe (33 yrs old) and I like to code."
+
+        # Act
+        response = llm.query_openai(
+            query=input_text,
+            response_model=dm.Person,
+            base_url=OLLAMA_HOST,
+            llm_model=OLLAMA_MODEL,
+            api_key="ollama",
+        )
+
+        # Assert
+        assert response