Use alias generator when deserializing the config file

This commit is contained in:
Saba 2021-11-28 13:05:48 -05:00
parent 19b81e82f0
commit 34d1e4199c
2 changed files with 12 additions and 11 deletions

View file

@ -1,22 +1,22 @@
content_type: content-type:
image: null image: null
ledger: null ledger: null
music: null music: null
org: org:
compressed_jsonl: .notes.json.gz compressed-jsonl: .notes.json.gz
embeddings_file: .note_embeddings.pt embeddings-file: .note_embeddings.pt
input_files: input-files:
- /home/saba/notes/notes.org - /home/saba/notes/notes.org
- /home/saba/notes/writing.org - /home/saba/notes/writing.org
input_filter: null input-filter: null
processor: processor:
conversation: conversation:
conversation_history: '' conversation-history: ''
conversation_logfile: .conversation_logs.json conversation-logfile: .conversation_logs.json
open_api_key: null open-api-key: null
search_type: search-type:
asymmetric: 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 encoder: sentence-transformers/msmarco-MiniLM-L-6-v3
image: image:
encoder: clip-ViT-B-32 encoder: clip-ViT-B-32

View file

@ -38,8 +38,9 @@ def config():
@app.post('/config') @app.post('/config')
async def config(updated_config: FullConfig): async def config(updated_config: FullConfig):
print(updated_config.dict())
with open(config_file, 'w') as outfile: 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() outfile.close()
return updated_config return updated_config