Merge pull request #979 from roninio/Azure-Embedding-Quota-Limit

Azure embedding quota limit
assafelovic · Nov 11, 2024 · 73d5750 · 73d5750
2 parents f363b6b + b2baadd
commit 73d5750
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/docs/docs/gpt-researcher/llms/llms.md b/docs/docs/gpt-researcher/llms/llms.md
@@ -65,6 +65,9 @@ See also the documentation in the Langchain [Azure OpenAI](https://api.python.la
 
 On Azure OpenAI you will need to create deployments for each model you want to use. Please also specify the model names/deployment names in your `.env` file:
 
+Required Embedding Model:
+To ensure optimal performance, GPT Researcher requires the text-embedding-3-large model. Please deploy this specific model to your Azure Endpoint.
+
 ```bash
 AZURE_OPENAI_API_KEY=[Your Key]
 AZURE_OPENAI_ENDPOINT=https://{your-endpoint}.openai.azure.com/
@@ -75,7 +78,7 @@ FAST_LLM=azure_openai:gpt-4o-mini
 SMART_LLM=azure_openai:gpt-4o
 STRATEGIC_LLM=azure_openai:o1-preview
 
-EMBEDDING="azure_openai:text-embedding-ada-002"
+
 ```
 
 

diff --git a/gpt_researcher/config/config.py b/gpt_researcher/config/config.py
@@ -67,9 +67,9 @@ def _handle_deprecated_attributes(self) -> None:
                 case "custom":
                     self.embedding_model = os.getenv("OPENAI_EMBEDDING_MODEL", "custom")
                 case "openai":
-                    self.embedding_model = "text-embedding-3-small"
+                    self.embedding_model = "text-embedding-3-large"
                 case "azure_openai":
-                    self.embedding_model = os.environ["AZURE_EMBEDDING_MODEL"]
+                    self.embedding_model = "text-embedding-3-large"
                 case "huggingface":
                     self.embedding_model = "sentence-transformers/all-MiniLM-L6-v2"
                 case _: