fix tests env var and lint

.github/workflows/pr_agent.yml

@@ -14,5 +14,5 @@ jobs:
         id: pragent
         uses: Codium-ai/pr-agent@main
         env:
-          OPENAI_KEY: ${{ secrets.OPENAI_KEY }}
+          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

semantic_router/utils/llm.py

@@ -33,6 +33,7 @@ def llm(prompt: str) -> str | None:
         logger.error(f"LLM error: {e}")
         raise Exception(f"LLM error: {e}")
 
+
 # TODO integrate async LLM function
 # async def allm(prompt: str) -> str | None:
 #     try:

tests/unit/test_layer.py

@@ -178,7 +178,10 @@ class TestRouteLayer:
         route_layer.to_json("test_output.json")
         assert os.path.exists("test_output.json")
         route_layer_from_file = RouteLayer.from_json("test_output.json")
-        assert route_layer_from_file.index is not None and route_layer_from_file.categories is not None
+        assert (
+            route_layer_from_file.index is not None
+            and route_layer_from_file.categories is not None
+        )
         os.remove("test_output.json")
 
     def test_yaml(self, openai_encoder, routes):
@@ -186,7 +189,10 @@ class TestRouteLayer:
         route_layer.to_yaml("test_output.yaml")
         assert os.path.exists("test_output.yaml")
         route_layer_from_file = RouteLayer.from_yaml("test_output.yaml")
-        assert route_layer_from_file.index is not None and route_layer_from_file.categories is not None
+        assert (
+            route_layer_from_file.index is not None
+            and route_layer_from_file.categories is not None
+        )
         os.remove("test_output.yaml")
 
     def test_config(self, openai_encoder, routes):
@@ -200,6 +206,7 @@ class TestRouteLayer:
         assert (route_layer_from_config.categories == route_layer.categories).all()
         assert route_layer_from_config.score_threshold == route_layer.score_threshold
 
+
 # Add more tests for edge cases and error handling as needed.