Skip to content

Commit

Permalink
[FIX][Better Testing System]
Browse files Browse the repository at this point in the history
  • Loading branch information
kyegomez committed Dec 17, 2024
1 parent 9abe300 commit e98d618
Show file tree
Hide file tree
Showing 7 changed files with 8 additions and 21 deletions.
2 changes: 1 addition & 1 deletion docs/mkdocs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -255,6 +255,7 @@ nav:
# - An Analysis on Prompting Strategies: "swarms/prompts/overview.md"
# - Managing Prompts in Production: "swarms/prompts/main.md"
- Community:
- Bounty Program: "corporate/bounty_program.md"
- Contributing:
- Contributing: "swarms/contributing.md"
- Tests: "swarms/framework/test.md"
Expand All @@ -264,7 +265,6 @@ nav:
- Swarms 5.6.8: "swarms/changelog/5_6_8.md"
- Swarms 5.8.1: "swarms/changelog/5_8_1.md"
- Swarms 5.9.2: "swarms/changelog/changelog_new.md"
- Bounty Program: "corporate/bounty_program.md"
- Corporate:
- Culture: "corporate/culture.md"
- Hiring: "corporate/hiring.md"
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ psutil = "*"
sentry-sdk = "*"
python-dotenv = "*"
PyYAML = "*"
docstring_parser = "0.16"
docstring_parser = "0.16" # TODO:
tiktoken = "*"
networkx = "*"
aiofiles = "*"
Expand Down
6 changes: 0 additions & 6 deletions swarms/structs/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,6 @@

import toml
import yaml

# from swarms.utils.loguru_logger import initialize_logger
from loguru import logger
from pydantic import BaseModel
from swarm_models.tiktoken_wrapper import TikTokenizer
Expand Down Expand Up @@ -500,10 +498,6 @@ def __init__(
if preset_stopping_token is not None:
self.stopping_token = "<DONE>"

# # Check the parameters
# # Telemetry Processor to log agent data
# threading.Thread(target=self.agent_initialization()).start

# If the docs exist then ingest the docs
if exists(self.docs):
threading.Thread(
Expand Down
1 change: 0 additions & 1 deletion swarms/structs/output_types.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from typing import Literal

# Literal of output types
# Literal of output types
OutputType = Literal[
"all",
Expand Down
8 changes: 0 additions & 8 deletions auto_test_eval.py → tests/agent_evals/auto_test_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,9 +92,6 @@ def __init__(
self.issues_created = []
self.last_issue_time = datetime.now()

# Validate GitHub token
# self._validate_github_credentials()

def _get_swarms_version(self) -> str:
"""Get the installed version of Swarms."""
try:
Expand Down Expand Up @@ -316,11 +313,6 @@ def report_swarms_issue(
return None


# from swarms import Agent
# from swarm_models import OpenAIChat
# from swarms.utils.issue_reporter import SwarmsIssueReporter
# import os

# Setup the reporter with your GitHub token
reporter = SwarmsIssueReporter(
github_token=os.getenv("GITHUB_API_KEY")
Expand Down
File renamed without changes.
10 changes: 6 additions & 4 deletions test_agent_features.py → tests/structs/test_agent_features.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
import asyncio
from swarms import Agent
from swarm_models import OpenAIChat
import json
import os
import tempfile
import time
import json

import yaml
import tempfile
from swarm_models import OpenAIChat

from swarms import Agent


def test_basic_agent_functionality():
Expand Down

0 comments on commit e98d618

Please sign in to comment.