Skip to content

Commit

Permalink
Update run_costorm_gpt.py
Browse files Browse the repository at this point in the history
  • Loading branch information
shaoyijia authored Sep 25, 2024
1 parent 0777607 commit 4c16d2e
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions examples/costorm_examples/run_costorm_gpt.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
"""
STORM Wiki pipeline powered by GPT-4o/4o-mini and Bing search engine.
Co-STORM pipeline powered by GPT-4o/4o-mini and Bing search engine.
You need to set up the following environment variables to run this script:
- OPENAI_API_KEY: OpenAI API key
- OPENAI_API_TYPE: OpenAI API type (e.g., 'openai' or 'azure')
Expand Down Expand Up @@ -116,6 +116,9 @@ def main(args):
# warm start the system
costorm_runner.warm_start()

# Below is an example of how users may interact with Co-STORM to seek information together
# In actual deployment, we suggest allowing the user to decide whether to observe the agent utterance or inject a turn

# observing Co-STORM LLM agent utterance for 5 turns
for _ in range(1):
conv_turn = costorm_runner.step()
Expand Down Expand Up @@ -235,4 +238,4 @@ def main(args):
help='If set, enable console log print.'
)

main(parser.parse_args())
main(parser.parse_args())

0 comments on commit 4c16d2e

Please sign in to comment.