Skip to content

Commit

Permalink
minor formatting prints
Browse files Browse the repository at this point in the history
  • Loading branch information
choccccy committed Jul 17, 2023
1 parent 4557d42 commit f02780d
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions demo/chat_web_selects.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@
```sh
python demo/chat_web_selects.py "www.newworldencyclopedia.org/entry/Igbo_People"
```
An example question might be "Who are the neighbors of the Igbo people?"
'''
# en.wikipedia.org/wiki/Igbo_people|ahiajoku.igbonet.com/2000/|en.wikivoyage.org/wiki/Igbo_phrasebook"
import asyncio
Expand All @@ -44,12 +46,12 @@
DOC_EMBEDDINGS_LLM = 'all-MiniLM-L6-v2'

COLLECTION_NAME = 'chat-web-selects'
USER_PROMPT = 'What do you want to know from these sites?:'
USER_PROMPT = 'What do you want to know from these sites?\n'

# Hard-code for demo
EMBED_CHUNK_SIZE = 200
EMBED_CHUNK_OVERLAP = 20
DOTS_SPACING = 0.5 # Number of seconds between each dot printed to console
DOTS_SPACING = 0.2 # Number of seconds between each dot printed to console


async def indicate_progress(pause=DOTS_SPACING):
Expand Down Expand Up @@ -99,6 +101,7 @@ async def async_main(sites, api_params):

done = False
while not done:
print()
user_question = input(USER_PROMPT)
if user_question == done:
break
Expand Down

0 comments on commit f02780d

Please sign in to comment.