SQL_Query_Completion/params.yaml at main · pranitab07/SQL_Query_Completion · GitHub

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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
base:
  project: sql_query_completion
  sleep_time: 0.2
  env_path: .env
  window_name:
    - MySQL Workbench
    - pgAdmin 4
data:
  loc: data/raw/sql_text_to_sql.csv
  source: gretelai/synthetic_text_to_sql

llm:
  model_name: llama-3.3-70b-versatile

triggers:
  initiater: ctrl+c
  filler: tab
  key: \t
  quiting: esc
  remove_ghost:
  c: ctrl
  key: z
  speed_write: 0.01

vector_store:
  type: pinecone
  path_to_save: data/processed/sql_faiss_index.index
  metadata_path: data/processed/sql_metadata.pkl
  top_k: 5 # for similarity search

  # only for chroma
  persist_directory: ./chroma
  collection_name_chroma: sql_collection
  chroma_batch_size: 100000

  # only for pinecone
  index_name: sql-index
  dimension: 384
  namespace: default
  metric: cosine         # or euclidean, dotproduct
  batch_size: 100

embedding_model: all-MiniLM-L6-v2

# mlflow
mlflow_config:
  api_delay: 5
  rate_limit_delay: 60
  sample_size: 100
  artifacts_dir: artifacts
  experiment_name: sql_models
  registered_model_name: LLama_3.0
  remote_server_uri: http://127.0.0.1:5000

memory:
  enable: true
  limit: 3