call gpt-4v, remove call_gpt

README.md

@@ -27,8 +27,8 @@ pip install -r requirements-dev.txt
 # Run camera tests (require a connected camera)
 pytest -m 'camera'
 
-# Run ChatGPT API tests (Warning: The ChatGPT API will be actually called. The API Key is required.)
-pytest -s -m 'chatgpt'
+# Run GPT API tests (Warning: The GPT API will be actually called. The API Key is required.)
+pytest -s -m 'gpt'
 
 # Run servo moter tests (require connected servo motors)
 pytest -m 'servo'

robot_gpt/robot.py

@@ -103,20 +103,6 @@ def call_gpt4v(self) -> str:
 
         return content
 
-    def call_gpt(self) -> str:
-        logger.info("Start OpenAI API call.")
-        logger.info(f"Prompts: {self.prompts}")
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=self.prompts
-        )
-        content = response.choices[0].message.content
-        logger.info(f"Response content: {content}")
-
-        self.append_prompt(Role.ASSISTANT, content)
-
-        return content
-
     def call_and_recognize(self):
         response = self.call_gpt4()
         json_response = json.loads(response)
@@ -131,4 +117,4 @@ def talk(self, message: str) -> str:
         self.append_prompt(Role.USER, message)
         self.append_prompt(Role.SYSTEM, "You can answer in free format only once.")
 
-        return self.call_gpt()
+        return self.call_gpt4v()

tests/test_robot.py

@@ -43,15 +43,14 @@ def test_recognize(mocker):
         assert last_prompt["role"] == Role.USER.value
 
 
-@ pytest.mark.chatgpt
+@ pytest.mark.gpt
 def test_call_gpt(mocker):
-    """Warning: The ChatGPT API will be actually called. The API Key is required."""
+    """Warning: The GPT API will be actually called. The API Key is required."""
     with RobotGPT() as robot:
         mocker.patch("robot_gpt.hardware.Camera.capture_image", return_value="./outputs/captured_image.jpg")
-        mocker.patch("robot_gpt.hardware.recognize_objects", return_value=["skeaker", "pcmonitor", "keyboard"])
         robot.recognize()
 
-        response = robot.call_gpt()
+        response = robot.call_gpt4v()
 
         # Response can be parsed in JSON
         json_response = json.loads(response)