Skip to content
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

feat: support bedrock embedding models #511

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

jrj5423
Copy link
Collaborator

@jrj5423 jrj5423 commented Dec 16, 2024

Close #379

Copy link

vercel bot commented Dec 16, 2024

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Comments Updated (UTC)
tidb-ai-preview ✅ Ready (Inspect) Visit Preview 💬 Add feedback Dec 23, 2024 2:24am
tidb-ai-storybook ✅ Ready (Inspect) Visit Preview 💬 Add feedback Dec 23, 2024 2:24am

Copy link

vercel bot commented Dec 16, 2024

@jrj5423 is attempting to deploy a commit to the pingcap Team on Vercel.

A member of the Team first needs to authorize it.

| ------------------------------- | ----------------- | ---------- |
| `amazon.titan-embed-text-v2:0` | 1024 | 8192 |
| `amazon.titan-embed-text-v1` | 1536 | 8192 |
| `amazon.titan-embed-g1-text-02` | 1536 | 8192 |
Copy link
Member

Choose a reason for hiding this comment

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

It seems AWS provides more models than these three?

For example cohere:xxxxx

Copy link
Collaborator Author

@jrj5423 jrj5423 Dec 20, 2024

Choose a reason for hiding this comment

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

Acknowledged, I have added all other embedding models supported by llama_index into the doc. I haven't tested those cohere models due to API key accessibility though.

Copy link
Collaborator

Choose a reason for hiding this comment

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

I suggest to document featured models and add a view more link to provider's document page.

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

Successfully merging this pull request may close these issues.

llm: support bedrock llm & embedding
4 participants