diff --git a/docs/configuration.md b/docs/configuration.md index 4dea64e6..537647e8 100644 --- a/docs/configuration.md +++ b/docs/configuration.md @@ -33,7 +33,7 @@ EMBEDDINGS_VECTOR_DIM=1024 If you want to use VoyageAI embeddings you will need to install `haiku.rag` with the VoyageAI extras, ```bash -uv pip install haiku.rag --extra voyageai +uv pip install haiku.rag[voyageai] ``` ```bash @@ -47,7 +47,7 @@ VOYAGE_API_KEY="your-api-key" If you want to use OpenAI embeddings you will need to install `haiku.rag` with the VoyageAI extras, ```bash -uv pip install haiku.rag --extra openai +uv pip install haiku.rag[openai] ``` and set environment variables. @@ -76,7 +76,7 @@ OLLAMA_BASE_URL="http://localhost:11434" For OpenAI QA, you need to install haiku.rag with OpenAI extras: ```bash -uv pip install haiku.rag --extra openai +uv pip install haiku.rag[openai] ``` Then configure: @@ -92,7 +92,7 @@ OPENAI_API_KEY="your-api-key" For Anthropic QA, you need to install haiku.rag with Anthropic extras: ```bash -uv pip install haiku.rag --extra anthropic +uv pip install haiku.rag[anthropic] ``` Then configure: @@ -125,7 +125,7 @@ RERANK_MODEL="mixedbread-ai/mxbai-rerank-base-v2" For Cohere reranking, install with Cohere extras: ```bash -uv pip install haiku.rag --extra cohere +uv pip install haiku.rag[cohere] ``` Then configure: diff --git a/docs/installation.md b/docs/installation.md index fd5eb509..cd6c21c8 100644 --- a/docs/installation.md +++ b/docs/installation.md @@ -15,19 +15,19 @@ For other embedding providers, install with extras: ### VoyageAI ```bash -uv pip install haiku.rag --extra voyageai +uv pip install haiku.rag[voyageai] ``` ### OpenAI ```bash -uv pip install haiku.rag --extra openai +uv pip install haiku.rag[openai] ``` ### Anthropic ```bash -uv pip install haiku.rag --extra anthropic +uv pip install haiku.rag[anthropic] ``` ## Requirements diff --git a/src/haiku/rag/embeddings/__init__.py b/src/haiku/rag/embeddings/__init__.py index eed18a75..bb371bd1 100644 --- a/src/haiku/rag/embeddings/__init__.py +++ b/src/haiku/rag/embeddings/__init__.py @@ -18,7 +18,7 @@ def get_embedder() -> EmbedderBase: raise ImportError( "VoyageAI embedder requires the 'voyageai' package. " "Please install haiku.rag with the 'voyageai' extra:" - "uv pip install haiku.rag --extra voyageai" + "uv pip install haiku.rag[voyageai]" ) return VoyageAIEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM) @@ -29,7 +29,7 @@ def get_embedder() -> EmbedderBase: raise ImportError( "OpenAI embedder requires the 'openai' package. " "Please install haiku.rag with the 'openai' extra:" - "uv pip install haiku.rag --extra openai" + "uv pip install haiku.rag[openai]" ) return OpenAIEmbedder(Config.EMBEDDINGS_MODEL, Config.EMBEDDINGS_VECTOR_DIM) diff --git a/src/haiku/rag/qa/__init__.py b/src/haiku/rag/qa/__init__.py index f9047a91..83e2451f 100644 --- a/src/haiku/rag/qa/__init__.py +++ b/src/haiku/rag/qa/__init__.py @@ -18,7 +18,7 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase: raise ImportError( "OpenAI QA agent requires the 'openai' package. " "Please install haiku.rag with the 'openai' extra:" - "uv pip install haiku.rag --extra openai" + "uv pip install haiku.rag[openai]" ) return QuestionAnswerOpenAIAgent(client, model or Config.QA_MODEL) @@ -29,7 +29,7 @@ def get_qa_agent(client: HaikuRAG, model: str = "") -> QuestionAnswerAgentBase: raise ImportError( "Anthropic QA agent requires the 'anthropic' package. " "Please install haiku.rag with the 'anthropic' extra:" - "uv pip install haiku.rag --extra anthropic" + "uv pip install haiku.rag[anthropic]" ) return QuestionAnswerAnthropicAgent(client, model or Config.QA_MODEL) diff --git a/src/haiku/rag/reranking/__init__.py b/src/haiku/rag/reranking/__init__.py index ccef8b9f..1a3920ca 100644 --- a/src/haiku/rag/reranking/__init__.py +++ b/src/haiku/rag/reranking/__init__.py @@ -29,7 +29,7 @@ def get_reranker() -> RerankerBase: raise ImportError( "Cohere reranker requires the 'cohere' package. " "Please install haiku.rag with the 'cohere' extra:" - "uv pip install haiku.rag --extra cohere" + "uv pip install haiku.rag[cohere]" ) _reranker = CohereReranker() return _reranker