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

22 lines
480 B
Python

from flow import create_embedding_flow
def main():
# Create the flow
flow = create_embedding_flow()
# Example text
text = "What's the meaning of life?"
# Prepare shared data
shared = {"text": text}
# Run the flow
flow.run(shared)
# Print results
print("Text:", text)
print("Embedding dimension:", len(shared["embedding"]))
print("First 5 values:", shared["embedding"][:5])
if __name__ == "__main__":
main()