config.yaml

models:
  sentence_transformer: "all-MiniLM-L6-v2"
  openai: "text-embedding-ada-002"

api_keys:
  openai: "your-openai-api-key"

chroma:
  collection_name: "documents"

faiss:
  dimension: 384  # Should match the SentenceTransformer model output dimension

fields:
  legal_department:
    prompt: "Which department should the completed form be returned to?"
    embedding_method: "sentence_transformer"
    top_k: 3
  party_info:
    prompt: "Who are the eligible parties for forwarding collateral?"
    embedding_method: "openai"
    top_k: 2
  contact_info:
    prompt: "What is the contact information for the trading desk?"
    embedding_method: "chroma"
    top_k: 1
  scope:
    prompt: "What is the scope of review mentioned?"
    embedding_method: "faiss"
    top_k: 2