Skip to content

enable end users to pass model instances for llm and embeddings model #128

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 2, 2024
Merged

Conversation

shkamboj1
Copy link
Contributor

@shkamboj1 shkamboj1 commented May 2, 2024

#123 Hi @VinciGit00, I have added support for passing model instances as well, means optionally end user can pass llm / embedding instances as well. So, using this we can support Azure OpenAI as well.

@shkamboj1 shkamboj1 changed the title end users to pass model instances for llm and embeddings model enable end users to pass model instances for llm and embeddings model May 2, 2024

"azure": {
"gpt-3.5-turbo": 4096
},
Copy link
Member

@lurenss lurenss May 2, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it would be cool to have all the avaible model on AzureAI, here the link Azure AI

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

sure.

@@ -92,6 +92,8 @@ def execute(self, state):
if isinstance(embedding_model, OpenAI):
embeddings = OpenAIEmbeddings(
api_key=embedding_model.openai_api_key)
elif isinstance(embedding_model, AzureOpenAIEmbeddings):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

DId you have the chanche to test with azure embedding model? Did you use text-ada? does it work properliy? :)

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

yes, I used text-ada, have attached example as well.

@VinciGit00 VinciGit00 merged commit 14eecef into ScrapeGraphAI:pre/beta May 2, 2024
Copy link

github-actions bot commented May 2, 2024

🎉 This PR is included in version 0.6.0 🎉

The release is available on:

Your semantic-release bot 📦🚀

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants