-
Notifications
You must be signed in to change notification settings - Fork 395
Expand file tree
/
Copy pathcustom_provider.yaml
More file actions
40 lines (36 loc) · 1.09 KB
/
Copy pathcustom_provider.yaml
File metadata and controls
40 lines (36 loc) · 1.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# This example demonstrates using a custom provider for RAG embedding models.
# For instance, you can use a local Ollama instance or any OpenAI-compatible
# API endpoint for generating embeddings.
providers:
local-ollama:
base_url: http://localhost:11434/v1
models:
local-embed:
provider: local-ollama
model: nomic-embed-text
agents:
root:
model: openai/gpt-5-mini
description: assistant with RAG using custom embedding provider
instruction: |
You are a helpful assistant with access to a knowledge base.
Use the search tool to find relevant information before answering.
toolsets:
- type: rag
ref: knowledge_base
rag:
knowledge_base:
tool:
description: search the knowledge base for relevant information
docs:
- ./docs
strategies:
- type: chunked-embeddings
embedding_model: local-embed # References the model defined above using the custom provider
database: ./custom_provider.db
vector_dimensions: 768
chunking:
size: 1000
overlap: 100
results:
limit: 5