Fixed tests

promptengineers-ai · Aug 13, 2023 · 1ee3d29 · 1ee3d29
1 parent f439dac
commit 1ee3d29
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/server/models/request.py b/server/models/request.py
@@ -10,7 +10,7 @@ class Message(BaseModel): # pylint: disable=too-few-public-methods
     messages: Optional[Any] = None
     temperature: Optional[float or int] = None
 
-    class Config: # pylint: disable=too-few-public-methods
+    class ConfigDict: # pylint: disable=too-few-public-methods
         """A message to send to the chatbot."""
         json_schema_extra = {
             "example": {

diff --git a/server/models/response.py b/server/models/response.py
@@ -3,7 +3,7 @@
 
 
 class ResponseStatus(BaseModel):
-    version: str = Field(example='v0.0.15')
+    version: str = Field(default='v0.0.15')
 
 #################################################
 ## ChatGPT
@@ -12,7 +12,7 @@ class ResponseStatus(BaseModel):
 class ResponseChat(BaseModel):
     # data: ChatCompletion
 
-	class Config: # pylint: disable=too-few-public-methods
+	class ConfigDict: # pylint: disable=too-few-public-methods
 		"""A message to send to the chatbot."""
 		json_schema_extra = {
 			"example": {

diff --git a/tests/integration/test_main.py b/tests/integration/test_main.py
@@ -1,6 +1,6 @@
 from starlette.testclient import TestClient
 
-from main import app
+from server.api import app
 
 client = TestClient(app)
 

diff --git a/tests/unit/test_example.py b/tests/unit/test_example.py
@@ -4,7 +4,7 @@
 
 class TestExampleCase(unittest.TestCase):
 
-    @unittest.skip("Example Test Case")
+    # @unittest.skip("Example Test Case")
     def test_retrieve_files(self):
         """Test that the files are retrieved."""
         token = "test"