You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
550 B
28 lines
550 B
from swarms.models import OpenAIChat
|
|
from swarms.structs import Flow
|
|
|
|
api_key = ""
|
|
|
|
|
|
# Initialize the language model,
|
|
# This model can be swapped out with Anthropic, ETC, Huggingface Models like Mistral, ETC
|
|
llm = OpenAIChat(
|
|
openai_api_key=api_key,
|
|
temperature=0.5,
|
|
max_tokens=100,
|
|
)
|
|
|
|
# Initialize the flow
|
|
flow = Flow(
|
|
llm=llm,
|
|
max_loops=5,
|
|
# system_prompt=SYSTEM_PROMPT,
|
|
# retry_interval=1,
|
|
)
|
|
|
|
out = flow.run("Generate a 10,000 word blog, say Stop when done")
|
|
print(out)
|
|
|
|
# # Now save the flow
|
|
# flow.save("flow.yaml")
|