Give direct access to the openai client

bakebrain · Jul 20, 2023 · caa315c · caa315c
1 parent ff8a4f7
commit caa315c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dsp/modules/gpt3.py b/dsp/modules/gpt3.py
@@ -55,7 +55,7 @@ def __init__(
 
         if api_key:
             openai.api_key = api_key
-        
+
         if kwargs.get("api_base"):
             openai.api_base = kwargs["api_base"]
 
@@ -72,6 +72,9 @@ def __init__(
             self.kwargs["model"] = model
         self.history: list[dict[str, Any]] = []
 
+    def _openai_client():
+        return openai
+
     def basic_request(self, prompt: str, **kwargs) -> OpenAIObject:
         raw_kwargs = kwargs