Fix lint errors, ported client.embeddings.create

llm/default_plugins/openai_models.py

@@ -101,12 +101,12 @@ def embed_batch(self, items: Iterable[Union[str, bytes]]) -> Iterator[List[float
         kwargs = {
             "input": items,
             "model": self.openai_model_id,
-            "api_key": self.get_key(),
         }
         if self.dimensions:
             kwargs["dimensions"] = self.dimensions
-        results = client.Embedding.create(**kwargs)["data"]
-        return ([float(r) for r in result["embedding"]] for result in results)
+        client = openai.OpenAI(api_key=self.get_key())
+        results = client.embeddings.create(**kwargs).data
+        return ([float(r) for r in result.embedding] for result in results)
 
 
 @hookimpl

llm/utils.py

@@ -25,7 +25,7 @@ def dicts_to_table_string(
     return res
 
 
-def remove_dict_none_values(d: dict) -> dict:
+def remove_dict_none_values(d):
     """
     Recursively remove keys with value of None or value of a dict that is all values of None
     """

tests/conftest.py

@@ -4,7 +4,7 @@
 import llm
 from llm.plugins import pm
 from pydantic import Field
-from pytest_httpx import HTTPXMock, IteratorStream
+from pytest_httpx import IteratorStream
 from typing import Optional