Skip to content

Commit e9873d2

Browse files
committed
more cleanup
1 parent 456ee9e commit e9873d2

File tree

2 files changed

+3
-24
lines changed

2 files changed

+3
-24
lines changed

amadeusgpt/main.py

Lines changed: 2 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -14,31 +14,10 @@
1414
SelfDebugLLM, VisualLLM)
1515
from amadeusgpt.integration_module_hub import IntegrationModuleHub
1616

17-
amadeus_fac = {}
18-
19-
20-
# using the config file to cache the amadeus instance
21-
# not sure if this is the best practice
22-
def create_amadeus(config: Config):
23-
if str(config) not in amadeus_fac:
24-
amadeus_fac[str(config)] = AMADEUS(config)
25-
return amadeus_fac[str(config)]
26-
2717
from amadeusgpt.analysis_objects.llm import (CodeGenerationLLM, DiagnosisLLM,
2818
SelfDebugLLM, VisualLLM)
2919
from amadeusgpt.integration_module_hub import IntegrationModuleHub
3020

31-
amadeus_fac = {}
32-
33-
34-
# using the config file to cache the amadeus instance
35-
# not sure if this is the best practice
36-
def create_amadeus(config: Config):
37-
if str(config) not in amadeus_fac:
38-
amadeus_fac[str(config)] = AMADEUS(config)
39-
return amadeus_fac[str(config)]
40-
41-
4221
class AMADEUS:
4322
def __init__(self, config: Dict[str, Any]):
4423
self.config = config
@@ -107,11 +86,11 @@ def run_task_program(self, task_program_name: str):
10786
if __name__ == "__main__":
10887
from amadeusgpt.analysis_objects.llm import VisualLLM
10988
from amadeusgpt.config import Config
110-
from amadeusgpt.main import create_amadeus
89+
11190

11291
config = Config("amadeusgpt/configs/EPM_template.yaml")
11392

114-
amadeus = create_amadeus(config)
93+
amadeus = AMADEUS(config)
11594
sandbox = amadeus.sandbox
11695
visualLLm = VisualLLM(config)
11796
visualLLm.speak(sandbox)

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ name = "AmadeusGPT"
77
readme = "README.md"
88
requires-python = ">=3.10"
99
dynamic = ["version"]
10-
version = "0.0.3"
10+
version = "0.1.0"
1111

1212
[tool.setuptools]
1313
packages = ["amadeusgpt"]

0 commit comments

Comments
 (0)