Cohere embeddings

import chromadb
from chromadb.utils.embedding_functions import CohereEmbeddingFunction
import os

# Set your API key
os.environ["COHERE_API_KEY"] = "your-api-key"

# Create embedding function
embedding_function = CohereEmbeddingFunction(
    api_key=os.environ["COHERE_API_KEY"],
    model_name="embed-english-v3.0"
)

# Create client and collection
client = chromadb.Client()
collection = client.create_collection(
    name="cohere_embeddings",
    embedding_function=embedding_function
)

# Add documents
collection.add(
    documents=["This is a document", "Another document"],
    ids=["doc1", "doc2"]
)

# Query
results = collection.query(
    query_texts=["search query"],
    n_results=2
)

# For search documents
embedding_function = CohereEmbeddingFunction(
    api_key=os.environ["COHERE_API_KEY"],
    model_name="embed-english-v3.0",
    input_type="search_document"  # Optimize for document indexing
)

collection = client.create_collection(
    name="documents",
    embedding_function=embedding_function
)

# Add documents with optimized embeddings
collection.add(
    documents=["Document content"],
    ids=["doc1"]
)

# For search queries, use search_query input type
query_embedding_function = CohereEmbeddingFunction(
    api_key=os.environ["COHERE_API_KEY"],
    model_name="embed-english-v3.0",
    input_type="search_query"  # Optimize for queries
)

# Create multilingual collection
embedding_function = CohereEmbeddingFunction(
    model_name="embed-multilingual-v3.0"
)

collection = client.create_collection(
    name="multilingual_docs",
    embedding_function=embedding_function
)

# Add documents in different languages
collection.add(
    documents=[
        "Hello world",  # English
        "Hola mundo",   # Spanish
        "Bonjour le monde",  # French
        "こんにちは世界"  # Japanese
    ],
    ids=["en", "es", "fr", "ja"]
)

# Query in any language
results = collection.query(
    query_texts=["greetings"],
    n_results=4
)

import { ChromaClient } from "chromadb";
import { CohereEmbeddingFunction } from "chromadb";

const embedder = new CohereEmbeddingFunction({
  cohere_api_key: "your-api-key",
  model: "embed-english-v3.0"
});

const client = new ChromaClient();
const collection = await client.createCollection({
  name: "cohere_collection",
  embeddingFunction: embedder
});

// Add and query documents
await collection.add({
  ids: ["id1", "id2"],
  documents: ["Document 1", "Document 2"]
});

const results = await collection.query({
  queryTexts: ["search query"],
  nResults: 2
});

import time
from cohere.errors import RateLimitError, AuthenticationError

try:
    collection.add(
        documents=["document"],
        ids=["id1"]
    )
except RateLimitError:
    # Handle rate limit
    time.sleep(60)  # Wait before retrying
except AuthenticationError:
    print("Invalid API key")