Merge pull request #26 from stacklok/fix-deprecation-errors

Update python-jsonlogger imports and pytest asyncio config

Author: jhrozek

pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+asyncio_default_fixture_loop_scope = function 

src/mockllm/config.py

@@ -6,10 +6,10 @@
 
 import yaml
 from fastapi import HTTPException
-from pythonjsonlogger import jsonlogger
+from pythonjsonlogger.json import JsonFormatter
 
 log_handler = logging.StreamHandler()
-log_handler.setFormatter(jsonlogger.JsonFormatter())
+log_handler.setFormatter(JsonFormatter())
 logging.basicConfig(level=logging.INFO, handlers=[log_handler])
 logger = logging.getLogger(__name__)
 
@@ -53,7 +53,7 @@ def load_responses(self) -> None:
     def get_response(self, prompt: str) -> str:
         """Get response for a given prompt."""
         self.load_responses()  # Check for updates
-        return self.responses.get(prompt , self.default_response)
+        return self.responses.get(prompt, self.default_response)
 
     def get_streaming_response(
         self, prompt: str, chunk_size: Optional[int] = None

src/mockllm/server.py

@@ -4,7 +4,7 @@
 import tiktoken
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import StreamingResponse
-from pythonjsonlogger import jsonlogger
+from pythonjsonlogger.json import JsonFormatter
 
 from .config import ResponseConfig
 from .models import (
@@ -20,7 +20,7 @@
 from .providers.openai import OpenAIProvider
 
 log_handler = logging.StreamHandler()
-log_handler.setFormatter(jsonlogger.JsonFormatter())
+log_handler.setFormatter(JsonFormatter())
 logging.basicConfig(level=logging.INFO, handlers=[log_handler])
 logger = logging.getLogger(__name__)