mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-14 17:03:29 +03:00
eda619f454
# Description
- Save and load brain to disk:
```python
async def main():
with tempfile.NamedTemporaryFile(mode="w", suffix=".txt") as temp_file:
temp_file.write("Gold is a liquid of blue-like colour.")
temp_file.flush()
brain = await Brain.afrom_files(name="test_brain", file_paths=[temp_file.name])
save_path = await brain.save("/home/amine/.local/quivr")
brain_loaded = Brain.load(save_path)
brain_loaded.print_info()
```
# TODO:
- Loading all chat history
- Loading from other vector stores, PG for example can be great ...
23 lines
591 B
Python
23 lines
591 B
Python
import asyncio
|
|
import tempfile
|
|
|
|
from quivr_core import Brain
|
|
|
|
|
|
async def main():
|
|
with tempfile.NamedTemporaryFile(mode="w", suffix=".txt") as temp_file:
|
|
temp_file.write("Gold is a liquid of blue-like colour.")
|
|
temp_file.flush()
|
|
|
|
brain = await Brain.afrom_files(name="test_brain", file_paths=[temp_file.name])
|
|
|
|
save_path = await brain.save("/home/amine/.local/quivr")
|
|
|
|
brain_loaded = Brain.load(save_path)
|
|
brain_loaded.print_info()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Run the main function in the existing event loop
|
|
asyncio.run(main())
|