Skip to content

Commit

Permalink
[CLEANUP]
Browse files Browse the repository at this point in the history
  • Loading branch information
Kye Gomez authored and Kye Gomez committed Jul 8, 2024
1 parent 41bb9d8 commit bbbbc9e
Show file tree
Hide file tree
Showing 5 changed files with 395 additions and 35 deletions.
22 changes: 1 addition & 21 deletions example.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,4 @@
from swarms import Agent
from langchain_community.llms.anthropic import Anthropic

from swarms import Agent, Anthropic

def calculate_profit(revenue: float, expenses: float):
"""
Expand Down Expand Up @@ -30,24 +28,6 @@ def generate_report(company_name: str, profit: float):
return f"The profit for {company_name} is ${profit}."


def write_memory_to_rag(memory_name: str, memory: str):
"""
Writes the memory to the RAG model for fine-tuning.
Args:
memory_name (str): The name of the memory.
memory (str): The memory to be written to the RAG model.
"""
# Write the memory to the RAG model for fine-tuning
from playground.memory.chromadb_example import ChromaDB

db = ChromaDB(output_dir=memory_name)

db.add(memory)

return None


# Initialize the agent
agent = Agent(
agent_name="Accounting Assistant",
Expand Down
2 changes: 1 addition & 1 deletion json_log_cleanup.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,4 +58,4 @@ def cleanup_json_logs(name: str = None):


# Call the function
cleanup_json_logs("sequential_workflow_agents")
cleanup_json_logs("social_media_swarm")
12 changes: 0 additions & 12 deletions playground/demos/social_media_content_generators_swarm/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,18 +12,6 @@

from swarms import Agent, OpenAIChat

# # Memory
# memory = ChromaDB(
# output_dir="social_media_marketing",
# docs_folder="docs",
# )

# Memory for instagram
# memory = ChromaDB(
# output_dir="social_media_marketing",
# docs_folder="docs",
# )

llm = OpenAIChat(max_tokens=4000)


Expand Down
Loading

0 comments on commit bbbbc9e

Please sign in to comment.