add TEST_USE_OPENAI env for running in github actions

.github/workflows/test.yaml

@@ -30,5 +30,6 @@ jobs:
     - name: "Test"
       env:
         OPENAI_API_KEY: ${{ secrets.openai_api_key }}
+        TEST_USE_OPENAI: "true"
       run: 'poetry run pytest'
       shell: bash

tests/ask_repo_test.py

@@ -4,16 +4,25 @@
 from ask_repo import AskRepo
 
 
-def test_invalid_repo_path():
-    with pytest.raises(FileNotFoundError):
-        AskRepo("/non_existent_path")
+class TestAskRepo:
+    @pytest.fixture
+    def model_name(self):
+        if os.getenv("TEST_USE_OPENAI"):
+            return "gpt-3.5-turbo"
+        else:
+            return "ollama/llama3.1:8b"
 
 
-def test_repo_map():
-    ask_repo = AskRepo(os.getcwd())
-    assert "pyproject.toml" in ask_repo.get_repo_map()
+    def test_invalid_repo_path(self):
+        with pytest.raises(FileNotFoundError):
+            AskRepo("/non_existent_path")
 
 
-def test_generates_valid_summary():
-    ask_repo = AskRepo(os.getcwd(), "gpt-3.5-turbo")
-    assert "summary" in ask_repo.summarise()
+    def test_repo_map(self):
+        ask_repo = AskRepo(os.getcwd())
+        assert "pyproject.toml" in ask_repo.get_repo_map()
+
+
+    def test_generates_valid_summary(self, model_name):
+        ask_repo = AskRepo(os.getcwd(), model_name)
+        assert "summary" in ask_repo.summarise()