mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-25 16:33:21 +08:00
29 lines
760 B
Plaintext
29 lines
760 B
Plaintext
name: rag_cot
|
|
python_env: python_env.yml
|
|
|
|
entry_points:
|
|
main:
|
|
parameters:
|
|
|
|
query:
|
|
description: Query to run
|
|
type: string
|
|
|
|
input_chromadb_artifact:
|
|
description: Fully-qualified name for the input artifact
|
|
type: string
|
|
|
|
embedding_model:
|
|
description: Fully-qualified name for the embedding model
|
|
type: string
|
|
|
|
chat_model_provider:
|
|
description: Fully-qualified name for the chat model provider
|
|
type: string
|
|
|
|
|
|
command: >-
|
|
python run.py --query {query} \
|
|
--input_chromadb_artifact {input_chromadb_artifact} \
|
|
--embedding_model {embedding_model} \
|
|
--chat_model_provider {chat_model_provider} |