diff --git a/config.yml b/config.yml index 701af2a2..c19f7a50 100644 --- a/config.yml +++ b/config.yml @@ -1,22 +1,22 @@ -content_type: +content-type: image: null ledger: null music: null org: - compressed_jsonl: .notes.json.gz - embeddings_file: .note_embeddings.pt - input_files: + compressed-jsonl: .notes.json.gz + embeddings-file: .note_embeddings.pt + input-files: - /home/saba/notes/notes.org - /home/saba/notes/writing.org - input_filter: null + input-filter: null processor: conversation: - conversation_history: '' - conversation_logfile: .conversation_logs.json - open_api_key: null -search_type: + conversation-history: '' + conversation-logfile: .conversation_logs.json + open-api-key: null +search-type: asymmetric: - cross_encoder: cross-encoder/ms-marco-MiniLM-L-6-v2 + cross-encoder: cross-encoder/ms-marco-MiniLM-L-6-v2 encoder: sentence-transformers/msmarco-MiniLM-L-6-v3 image: encoder: clip-ViT-B-32 diff --git a/src/main.py b/src/main.py index ee9803c2..ab8db7e2 100644 --- a/src/main.py +++ b/src/main.py @@ -38,8 +38,9 @@ def config(): @app.post('/config') async def config(updated_config: FullConfig): + print(updated_config.dict()) with open(config_file, 'w') as outfile: - yaml.dump(yaml.safe_load(updated_config.json()), outfile) + yaml.dump(yaml.safe_load(updated_config.json(by_alias=True)), outfile) outfile.close() return updated_config