Plugin AiRag

Search ​Search

yaml
type: "io.kestra.plugin.ai.rag.search"
yaml
id: search_embeddings_flow
namespace: company.ai

tasks:
  - id: ingest
    type: io.kestra.plugin.ai.rag.IngestDocument
    provider:
      type: io.kestra.plugin.ai.provider.GoogleGemini
      modelName: gemini-embedding-exp-03-07
      apiKey: "{{ kv('GEMINI_API_KEY') }}"
    embeddings:
      type: io.kestra.plugin.ai.embeddings.KestraKVStore
    drop: true
    fromExternalURLs:
      - https://raw.githubusercontent.com/kestra-io/docs/refs/heads/main/content/blogs/release-0-22.md

  - id: search
    type: io.kestra.plugin.ai.rag.Search
    provider:
      type: io.kestra.plugin.ai.provider.GoogleGemini
      modelName: gemini-embedding-exp-03-07
      apiKey: "{{ kv('GEMINI_API_KEY') }}"
    embeddings:
      type: io.kestra.plugin.ai.embeddings.KestraKVStore
    query: "Feature Highlights"
    maxResults: 5
    minScore: 0.5
    fetchType: FETCH
Properties
Default NONE
Possible Values
STOREFETCHFETCH_ONENONE
SubType string
Format uri
Default false
SubType string
Default {{flow.id}}-embedding-store
Default embedding-index
Default https://api.deepseek.com/v1
Possible Values
ONEQUORUMALL
SubType string
SubType string
Min items 1
SubType string
Default COHERE
Possible Values
COHERETITAN