Merge pull request #22 from rawkintrevo/21

Closes #21 - Only include specified tools

gofannon/arxiv/__init__.py

@@ -1,2 +0,0 @@
-from.search import Search
-from.get_article import GetArticle  

gofannon/basic_math/__init__.py

@@ -1,5 +1 @@
-from.exponents import Exponents
-from.multiplication import Multiplication
-from.division import Division
-from.addition import Addition
-from.subtraction import Subtraction
+

gofannon/github/__init__.py

@@ -1,11 +1,4 @@
 
-from.create_issue import CreateIssue
-from.get_repo_contents import GetRepoContents
-from.commit_file import CommitFile
-from.commit_files import CommitFiles
-from.search import SearchRepos
-from .read_issue import ReadIssue
-
 
 
 

gofannon/reasoning/__init__.py

@@ -1,3 +0,0 @@
-from .sequential_cot import SequentialCoT
-from .hierarchical_cot import HierarchicalCoT
-from .tree_of_thought import TreeOfThought

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "gofannon"
-version = "0.1.0-prerelease"
+version = "0.1.0"
 description = "A collection of tools for LLMs"
 authors = ["Trevor Grant <[email protected]>"]
 license = "ASFv2"

tests/test_arxiv.py

@@ -1,5 +1,6 @@
 import pytest
-from gofannon.arxiv import GetArticle, Search
+from gofannon.arxiv.get_article import GetArticle
+from gofannon.arxiv.search import Search
 
 def test_get_article():
     get_article = GetArticle()

tests/test_basic_math.py

@@ -1,5 +1,9 @@
 import pytest
-from gofannon.basic_math import Addition, Subtraction, Multiplication, Division, Exponents
+from gofannon.basic_math.addition import Addition
+from gofannon.basic_math.subtraction import Subtraction
+from gofannon.basic_math.multiplication import Multiplication
+from gofannon.basic_math.division import Division
+from gofannon.basic_math.exponents import Exponents
 
 def test_addition():
     addition = Addition()

tests/test_github.py

@@ -1,6 +1,6 @@
 import pytest
 from os import getenv
-from gofannon.github import CommitFile, CreateIssue, GetRepoContents, ReadIssue, SearchRepos
+# from gofannon.github import CommitFile, CreateIssue, GetRepoContents, ReadIssue, SearchRepos
 
 # def test_commit_file():
 #     commit_file = CommitFile(api_key=getenv('TEST_GITHUB_TOKEN'))

tests/test_integrations.py

@@ -2,7 +2,7 @@
 
 import pytest
 from gofannon.base import BaseTool
-from gofannon.basic_math import Addition
+from gofannon.basic_math.addition import Addition
 
 # Add DummyTool subclass implementing abstract methods  
 class DummyTool(BaseTool):

tests/test_reasoning.py

@@ -1,6 +1,8 @@
 import pytest
 from os import getenv
-from gofannon.reasoning import HierarchicalCoT, SequentialCoT, TreeOfThought
+from gofannon.reasoning.hierarchical_cot import HierarchicalCoT
+from gofannon.reasoning.sequential_cot import SequentialCoT
+from gofannon.reasoning.tree_of_thought import TreeOfThought
 depth_chart = [
     {'model_name' : "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
      'base_url' : "https://api.deepinfra.com/v1/openai",