Update examples to work with the separate controller and lm modules

examples/doc_merge/doc_merge.py

@@ -14,7 +14,7 @@ import json
 import csv
 from statistics import fmean
 from typing import Dict, List, Callable, Set, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 
 class DocMergePrompter(prompter.Prompter):
@@ -714,8 +714,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/keyword_counting/keyword_counting.py

@@ -18,7 +18,7 @@ import csv
 from collections import Counter
 from functools import partial
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 
 def string_to_list(string: str) -> List[str]:
@@ -1403,8 +1403,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/set_intersection/set_intersection_032.py

@@ -15,7 +15,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 from . import utils
 
@@ -672,8 +672,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/set_intersection/set_intersection_064.py

@@ -15,7 +15,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 from . import utils
 
@@ -702,8 +702,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/set_intersection/set_intersection_128.py

@@ -15,7 +15,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 from . import utils
 
@@ -754,8 +754,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/sorting/sorting_032.py

@@ -12,7 +12,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 from . import utils
 
 
@@ -673,8 +673,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/sorting/sorting_064.py

@@ -12,7 +12,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 from . import utils
 
@@ -732,8 +732,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )

examples/sorting/sorting_128.py

@@ -12,7 +12,7 @@ import datetime
 import json
 import csv
 from typing import Dict, List, Callable, Union
-from graph_of_thoughts import controller, operations, prompter, parser
+from graph_of_thoughts import controller, language_models, operations, prompter, parser
 
 from . import utils
 
@@ -830,8 +830,8 @@ def run(
                     f"Budget has been depleted, stopping. Method {method.__name__} has not been run."
                 )
                 break
-            lm = controller.ChatGPT(
+            lm = language_models.ChatGPT(
-                "../../graph_of_thoughts/controller/config.json",
+                "../../graph_of_thoughts/language_models/config.json",
                 model_name=lm_name,
                 cache=True,
             )