Skip to content

Commit

Permalink
quick toml update
Browse files Browse the repository at this point in the history
  • Loading branch information
jonnyjohnson1 committed Jul 8, 2024
1 parent fabe1aa commit 10d28da
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 2 deletions.
Binary file added _conv_cache/cache__HiAIB1Ag6PFn.pkl
Binary file not shown.
1 change: 1 addition & 0 deletions config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
active_spacy_model: en_core_web_trf
4 changes: 4 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ emoji = "0.6.0"
pyyaml = "6.0.1"
python-dotenv = "1.0.0"
openai = "1.30.4"
<<<<<<< Updated upstream
neo4j = "^5.20.0"
nltk = "^3.8.1"
scikit-learn = "^1.5.0"
Expand All @@ -38,6 +39,9 @@ pyjwt = "^2.8.0"
python-multipart = "^0.0.9"
pytest-asyncio = "^0.23.7"
textblob = "^0.18.0.post0"
=======
tk = "0.1.0"
>>>>>>> Stashed changes

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.3"
Expand Down
12 changes: 11 additions & 1 deletion topos/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

def main():
parser = argparse.ArgumentParser(description="CLI for my script")
parser.add_argument('command', choices=['run', 'set', 'chat'], help="Command to execute")
parser.add_argument('command', choices=['run', 'set', 'chat', 'zrok'], help="Command to execute")
parser.add_argument('--web', action='store_true', help="Flag to run the server for web access")
parser.add_argument('--local', action='store_true', help="Flag to run the server for local access (default)")
parser.add_argument('--spacy', choices=['small', 'med', 'large', 'trf'], help="Specify Spacy model size (only for 'set' command)")
Expand All @@ -30,6 +30,16 @@ def main():
from .chat_api import api
api.start_chat()

if args.command == 'zrok':
"""
start the topos api chat server for clients to connect
"""
# start zrok server
import subprocess
# zrok share public http://0.0.0.0:13341 # the cli command
subprocess.run(['zrok', 'share', 'public', 'http://0.0.0.0:13341'], check=True)


elif args.command == 'set':
"""
download Spacy model
Expand Down
2 changes: 1 addition & 1 deletion topos/config.yaml
Original file line number Diff line number Diff line change
@@ -1 +1 @@
active_spacy_model: en_core_web_md
active_spacy_model: en_core_web_trf

0 comments on commit 10d28da

Please sign in to comment.