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.
24 lines
609 B
24 lines
609 B
from swarms.structs.swarm_matcher import (
|
|
SwarmMatcher,
|
|
SwarmMatcherConfig,
|
|
)
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# Example usage
|
|
if __name__ == "__main__":
|
|
# Create configuration
|
|
config = SwarmMatcherConfig(
|
|
backend="openai", # Using local embeddings for this example
|
|
similarity_threshold=0.6, # Increase threshold for more strict matching
|
|
cache_embeddings=True,
|
|
)
|
|
|
|
# Initialize matcher
|
|
matcher = SwarmMatcher(config)
|
|
|
|
task = "I need to build a hierarchical swarm of agents to solve a problem"
|
|
|
|
print(matcher.auto_select_swarm(task))
|