Skip to content
Snippets Groups Projects
Commit 15f8af84 authored by maxyousif15's avatar maxyousif15
Browse files

feat: adding support for open ai org id for the openai encoder

parent a3732f1f
No related branches found
No related tags found
No related merge requests found
...@@ -20,6 +20,7 @@ class OpenAIEncoder(BaseEncoder): ...@@ -20,6 +20,7 @@ class OpenAIEncoder(BaseEncoder):
self, self,
name: Optional[str] = None, name: Optional[str] = None,
openai_api_key: Optional[str] = None, openai_api_key: Optional[str] = None,
openai_org_id: Optional[str] = None,
score_threshold: float = 0.82, score_threshold: float = 0.82,
dimensions: Union[int, NotGiven] = NotGiven(), dimensions: Union[int, NotGiven] = NotGiven(),
): ):
...@@ -27,10 +28,13 @@ class OpenAIEncoder(BaseEncoder): ...@@ -27,10 +28,13 @@ class OpenAIEncoder(BaseEncoder):
name = os.getenv("OPENAI_MODEL_NAME", "text-embedding-ada-002") name = os.getenv("OPENAI_MODEL_NAME", "text-embedding-ada-002")
super().__init__(name=name, score_threshold=score_threshold) super().__init__(name=name, score_threshold=score_threshold)
api_key = openai_api_key or os.getenv("OPENAI_API_KEY") api_key = openai_api_key or os.getenv("OPENAI_API_KEY")
if api_key is None: print(f"api key: {api_key}")
org_id = openai_org_id or os.getenv("OPENAI_ORGANIZATION")
print(f"org id: {org_id}")
if (api_key is None) and (org_id is None):
raise ValueError("OpenAI API key cannot be 'None'.") raise ValueError("OpenAI API key cannot be 'None'.")
try: try:
self.client = openai.Client(api_key=api_key) self.client = openai.Client(api_key=api_key, organization=org_id)
except Exception as e: except Exception as e:
raise ValueError( raise ValueError(
f"OpenAI API client failed to initialize. Error: {e}" f"OpenAI API client failed to initialize. Error: {e}"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment