Skip to content

- use different models for extractors of audio/video #1

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.

Sign up for GitHub

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

Merged
weoses merged 1 commit into from
May 20, 2026
Merged

- use different models for extractors of audio/video #1

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions docker-compose.deploy.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,9 @@ services:

GEMINI_EXTRACTOR_APIKEY: ${GEMINI_EXTRACTOR_APIKEY}
GEMINI_EXTRACTOR_APIENDPOINT: ${GEMINI_EXTRACTOR_APIENDPOINT}
GEMINI_EXTRACTOR_MODEL: ${GEMINI_EXTRACTOR_MODEL}
GEMINI_EXTRACTOR_MODEL_IMAGE: ${GEMINI_EXTRACTOR_MODEL_IMAGE}
GEMINI_EXTRACTOR_MODEL_VIDEO: ${GEMINI_EXTRACTOR_MODEL_VIDEO}
GEMINI_EXTRACTOR_MODEL_AUDIO: ${GEMINI_EXTRACTOR_MODEL_AUDIO}

EXTRACTOR_PROVIDER: ${EXTRACTOR_PROVIDER}
EMBEDDER_PROVIDER: ${EMBEDDER_PROVIDER}
Expand All @@ -37,7 +39,11 @@ services:
OPENROUTER_EMBEDDING_MODEL: ${OPENROUTER_EMBEDDING_MODEL}

OPENROUTER_EXTRACTOR_APIKEY: ${OPENROUTER_EXTRACTOR_APIKEY}
OPENROUTER_EXTRACTOR_MODEL: ${OPENROUTER_EXTRACTOR_MODEL}
OPENROUTER_EXTRACTOR_MODEL_IMAGE: ${OPENROUTER_EXTRACTOR_MODEL_IMAGE}
OPENROUTER_EXTRACTOR_MODEL_VIDEO: ${OPENROUTER_EXTRACTOR_MODEL_VIDEO}
OPENROUTER_EXTRACTOR_MODEL_AUDIO: ${OPENROUTER_EXTRACTOR_MODEL_AUDIO}

EXTRACTING_SEPARATE_AUDIO: ${EXTRACTING_SEPARATE_AUDIO}

FFMPEG_BINARY: ${FFMPEG_BINARY}
FFMPEG_CPULIMIT: ${FFMPEG_CPULIMIT}
Expand Down
17 changes: 11 additions & 6 deletions group_vars/all.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,23 +14,28 @@ ffmpeg_binary: ffmpeg
ffmpeg_cpulimit: 80
ffmpeg_threadslimit:

# LLM provider selection (gemini | openrouter)
extractor_provider: "openrouter"
embedder_provider: "gemini"
extracting_separate_audio: false

# Gemini embedding model
gemini_embedding_model: "gemini-embedding-2-preview"
gemini_embedding_api_endpoint: ""

# Gemini extractor model
gemini_extractor_model: "gemini-2.5-flash"
gemini_extractor_model_image: "gemini-2.5-flash"
gemini_extractor_model_video: "gemini-2.5-flash"
gemini_extractor_model_audio: "gemini-2.5-flash"
gemini_extractor_api_endpoint: ""

# LLM provider selection (gemini | openrouter)
extractor_provider: "openrouter"
embedder_provider: "gemini"

# OpenRouter embedding model
openrouter_embedding_model: ""

# OpenRouter extractor model
openrouter_extractor_model: "google/gemini-3-flash-preview"
openrouter_extractor_model_image: "google/gemini-3-flash-preview"
openrouter_extractor_model_video: "google/gemini-3-flash-preview"
openrouter_extractor_model_audio: "google/gemini-3-flash-preview"


# filled by runner
Expand Down
9 changes: 7 additions & 2 deletions roles/deploy/templates/.env.j2
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ S3_SECURE={{ s3_secure }}
S3_MEDIA_BUCKET={{ s3_media_bucket }}
S3_TEMP_BUCKET={{ s3_temp_bucket }}

EXTRACTING_SEPARATE_AUDIO: {{ extracting_separate_audio }}

# Gemini embedding
GEMINI_EMBEDDING_APIKEY={{ gemini_api_key }}
Expand All @@ -24,7 +25,9 @@ GEMINI_EMBEDDING_MODEL={{ gemini_embedding_model }}
# Gemini extractor
GEMINI_EXTRACTOR_APIKEY={{ gemini_api_key }}
GEMINI_EXTRACTOR_APIENDPOINT={{ gemini_extractor_api_endpoint }}
GEMINI_EXTRACTOR_MODEL={{ gemini_extractor_model }}
GEMINI_EXTRACTOR_MODEL_IMAGE={{ gemini_extractor_model_image }}
GEMINI_EXTRACTOR_MODEL_VIDEO={{ gemini_extractor_model_video }}
GEMINI_EXTRACTOR_MODEL_AUDIO={{ gemini_extractor_model_audio }}

# LLM provider selection
EXTRACTOR_PROVIDER={{ extractor_provider }}
Expand All @@ -36,7 +39,9 @@ OPENROUTER_EMBEDDING_MODEL={{ openrouter_embedding_model }}

# OpenRouter extractor
OPENROUTER_EXTRACTOR_APIKEY={{ openrouter_api_key | default('') }}
OPENROUTER_EXTRACTOR_MODEL={{ openrouter_extractor_model }}
OPENROUTER_EXTRACTOR_MODEL_IMAGE={{ openrouter_extractor_model_image }}
OPENROUTER_EXTRACTOR_MODEL_VIDEO={{ openrouter_extractor_model_video }}
OPENROUTER_EXTRACTOR_MODEL_AUDIO={{ openrouter_extractor_model_audio }}

# PostgreSQL
POSTGRES_DB={{ postgres_db }}
Expand Down