Skip to content

Commit

Permalink
clean up of useless code, code, no more worker, etc
Browse files Browse the repository at this point in the history
  • Loading branch information
kyegomez committed Nov 7, 2023
1 parent 7bc6b9c commit bb496f4
Show file tree
Hide file tree
Showing 6 changed files with 8 additions and 65 deletions.
8 changes: 4 additions & 4 deletions groupchat.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,28 +14,28 @@
flow1 = Flow(
llm=llm,
max_loops=1,
system_message="YOU ARE SILLY, YOU OFFER NOTHING OF VALUE",
system_prompt="YOU ARE SILLY, YOU OFFER NOTHING OF VALUE",
name="silly",
dashboard=True,
)
flow2 = Flow(
llm=llm,
max_loops=1,
system_message="YOU ARE VERY SMART AND ANSWER RIDDLES",
system_prompt="YOU ARE VERY SMART AND ANSWER RIDDLES",
name="detective",
dashboard=True,
)
flow3 = Flow(
llm=llm,
max_loops=1,
system_message="YOU MAKE RIDDLES",
system_prompt="YOU MAKE RIDDLES",
name="riddler",
dashboard=True,
)
manager = Flow(
llm=llm,
max_loops=1,
system_message="YOU ARE A GROUP CHAT MANAGER",
system_prompt="YOU ARE A GROUP CHAT MANAGER",
name="manager",
dashboard=True,
)
Expand Down
2 changes: 1 addition & 1 deletion playground/workflow.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from swarms import Workflow
from swarms.tools.autogpt import ChatOpenAI
from swarms.models import ChatOpenAI

workflow = Workflow(ChatOpenAI)

Expand Down
1 change: 0 additions & 1 deletion swarms/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@


from swarms.workers import *
from swarms.workers.worker import Worker
from swarms.chunkers import *
from swarms.models import * # import * only works when __all__ = [] is defined in __init__.py
from swarms.structs import *
Expand Down
4 changes: 3 additions & 1 deletion swarms/models/biogpt.py
Original file line number Diff line number Diff line change
Expand Up @@ -105,12 +105,14 @@ def __call__(self, text: str):
generator = pipeline(
"text-generation", model=self.model, tokenizer=self.tokenizer
)
return generator(
out = generator(
text,
max_length=self.max_length,
num_return_sequences=self.num_return_sequences,
do_sample=self.do_sample,
)

return out[0]['generated_text']

def get_features(self, text):
"""
Expand Down
22 changes: 0 additions & 22 deletions swarms/tools/exit_conversation.py

This file was deleted.

36 changes: 0 additions & 36 deletions swarms/tools/requests.py

This file was deleted.

0 comments on commit bb496f4

Please sign in to comment.