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
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

View file

@ -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