pocketflow/cookbook/pocketflow-tool-search/main.py

27 lines
514 B
Python

import os
from flow import create_flow
def main():
"""Run the web search flow"""
# Get search query from user
query = input("Enter search query: ")
if not query:
print("Error: Query is required")
return
# Initialize shared data
shared = {
"query": query,
"num_results": 5
}
# Create and run flow
flow = create_flow()
flow.run(shared)
# Results are in shared["analysis"]
if __name__ == "__main__":
main()