Skip to content

Commit

Permalink
Moved algorithms to the algorithms folder, renamed algorithms.py to b…
Browse files Browse the repository at this point in the history
…ase.py
  • Loading branch information
NotBioWaste905 committed Nov 6, 2024
1 parent de6ee20 commit 25da331
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -71,5 +71,10 @@ class GraphGenerator(BaseAlgorithm):
def __init__(self):
super().__init__()

def invoke(self, dialogue: Dialogue, graph: BaseGraph = None, topic: str = "") -> BaseGraph:
def invoke(self, dialogue: Dialogue = None, graph: BaseGraph = None, topic: str = "") -> BaseGraph:
raise NotImplementedError


class CycleGraphGenerator(GraphGenerator):
def invoke(self, dialogue: Dialogue = None, graph: BaseGraph = None, topic: str = "") -> BaseGraph:

Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from chatsky_llm_autoconfig.algorithms.base import DialogAugmentation
from chatsky_llm_autoconfig.graph import BaseGraph
from chatsky_llm_autoconfig.dialogue import Dialogue
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import random
from chatsky_llm_autoconfig.graph import BaseGraph
from chatsky_llm_autoconfig.algorithms import DialogueGenerator
from chatsky_llm_autoconfig.algorithms.base import DialogueGenerator
from chatsky_llm_autoconfig.dialogue import Dialogue
from chatsky_llm_autoconfig.autometrics.registry import AlgorithmRegistry

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from chatsky_llm_autoconfig.algorithms.base import GraphGenerator
from chatsky_llm_autoconfig.graph import BaseGraph
from chatsky_llm_autoconfig.dialogue import Dialogue

0 comments on commit 25da331

Please sign in to comment.