everything working again after cleanup

vibe_bot/tests/test_llama_wrapper.py

@@ -6,6 +6,7 @@ import base64
 import tempfile
 from io import BytesIO
 from pathlib import Path
+from typing import Any
 from unittest.mock import MagicMock, patch
 
 import numpy as np
@@ -106,24 +107,24 @@ EMBEDDING_SIMILARITY_LOW = 0.5
 
 def test_embeddings() -> None:
     """Test embedding similarity for similar and different texts."""
-    with patch("vibe_bot.llama_wrapper.openai.OpenAI") as mock_openai:
-        mock_horse_vec = [0.8] * 1024 + [0.6] * 1024
-        mock_horse_also_vec = [0.79] * 1024 + [0.61] * 1024
-        mock_donkey_vec = [-0.8] * 1024 + [-0.6] * 1024
+    mock_horse_vec = [0.8] * 1024 + [0.6] * 1024
+    mock_horse_also_vec = [0.79] * 1024 + [0.61] * 1024
+    mock_donkey_vec = [-0.8] * 1024 + [-0.6] * 1024
 
-        mock_response1 = MagicMock()
-        mock_response1.data = [MagicMock(embedding=mock_horse_vec)]
-        mock_response2 = MagicMock()
-        mock_response2.data = [MagicMock(embedding=mock_horse_also_vec)]
-        mock_response3 = MagicMock()
-        mock_response3.data = [MagicMock(embedding=mock_donkey_vec)]
-
-        mock_openai.return_value.embeddings.create.side_effect = [
-            mock_response1,
-            mock_response2,
-            mock_response3,
-        ]
+    def mock_post(*args: Any, **kwargs: Any) -> MagicMock:
+        json_data = kwargs.get("json", {})
+        text = json_data["input"][0]
+        if "horse" in text and "donkey" not in text and "also" not in text:
+            embedding_data = mock_horse_vec
+        elif "also" in text:
+            embedding_data = mock_horse_also_vec
+        else:
+            embedding_data = mock_donkey_vec
+        mock_resp = MagicMock()
+        mock_resp.json.return_value = {"data": [{"embedding": embedding_data}]}
+        return mock_resp
 
+    with patch("vibe_bot.llama_wrapper.requests.post", side_effect=mock_post):
         result1 = embedding(
             "this is a horse",
             openai_url=EMBEDDING_ENDPOINT,