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

Alter get_embedding so that it is compatible with cohere models #27

Merged
merged 1 commit into from
Aug 9, 2024

Conversation

spodgorny9
Copy link
Collaborator

No description provided.

Copy link
Member

@grantbuster grantbuster left a comment

Choose a reason for hiding this comment

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

LGTM. If we get many more embedding models we'll want to setup a separate class to handle the formatting for them.

@@ -394,7 +394,7 @@ class EnergyWizardPostgres(EnergyWizardBase):

This class is designed as follows:
Vector database: PostgreSQL database accessed using psycopg2.
Query Embedding: AWS titan using boto3
Query Embedding: AWS using boto3
LLM Application: GPT4 via Azure deployment
"""
EMBEDDING_MODEL = 'amazon.titan-embed-text-v1'
Copy link
Member

Choose a reason for hiding this comment

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

Do you update this in the forked repo?

@spodgorny9 spodgorny9 merged commit 265ceb2 into main Aug 9, 2024
12 checks passed
@spodgorny9 spodgorny9 deleted the sp/aws_cohere branch August 9, 2024 00:09
github-actions bot pushed a commit that referenced this pull request Aug 9, 2024
Alter get_embedding so that it is compatible with cohere models
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.

2 participants