Update Gemini to include the embedding function - Githubissues

intelligentnode / IntelliNode

Access the latest AI models like ChatGPT, LLaMA, Diffusion, Gemini Hugging face, and beyond through a unified prompt layer and performance evaluation

https://show.intellinode.ai

Apache License 2.0

213 stars 13 forks source link

Update Gemini to include the embedding function #94

Closed intelligentnode closed 5 months ago

intelligentnode commented 5 months ago

Update the GeminiAIWrapper to include embedding functions:

Check google api example to call embedding

Update the GeminiAIWrapper.test.js with extra unit test case for embeding - print the response to check the parsing format.
Update the RemoteEmbedModel.js by adding Gemini as providing, calling the related function.

Edit places:

SupportedEmbedModels.
initiate.
getEmbeddings.
Add a function inside EmbedInput called getGeminiInputs().

4- Update RemoteEmbedModel.test.js to include extra unit test case for Gemini provider.

Notes:

Update config.json to include the gemini embedding endpoint.