Spaces:
Sleeping
Sleeping
Merge branch 'main' of https://github.com/stefanches7/AI-assisted-Neuroimaging-harmonization
Browse files- agent.py +9 -2
- requirements.txt +1 -0
agent.py
CHANGED
|
@@ -13,14 +13,21 @@ class BIDSifierAgent:
|
|
| 13 |
load_dotenv()
|
| 14 |
|
| 15 |
if provider=="openai":
|
| 16 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
else:
|
| 18 |
lm = dspy.LM(f"{provider}/{model}", api_key="")
|
|
|
|
|
|
|
|
|
|
| 19 |
dspy.configure(lm=lm)
|
| 20 |
self.llm = lm
|
| 21 |
self.model = model or os.getenv("BIDSIFIER_MODEL", "gpt-4o-mini")
|
| 22 |
self.temperature = temperature
|
| 23 |
-
|
| 24 |
def _build_user_prompt(self, step: str, context: Dict[str, Any]) -> str:
|
| 25 |
dataset_xml = context.get("dataset_xml")
|
| 26 |
readme_text = context.get("readme_text")
|
|
|
|
| 13 |
load_dotenv()
|
| 14 |
|
| 15 |
if provider=="openai":
|
| 16 |
+
if model == "gpt-5":
|
| 17 |
+
temperature = 1.0
|
| 18 |
+
lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"), temperature = temperature, max_tokens = 40000)
|
| 19 |
+
else:
|
| 20 |
+
lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"), temperature = temperature)
|
| 21 |
else:
|
| 22 |
lm = dspy.LM(f"{provider}/{model}", api_key="")
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
|
| 26 |
dspy.configure(lm=lm)
|
| 27 |
self.llm = lm
|
| 28 |
self.model = model or os.getenv("BIDSIFIER_MODEL", "gpt-4o-mini")
|
| 29 |
self.temperature = temperature
|
| 30 |
+
|
| 31 |
def _build_user_prompt(self, step: str, context: Dict[str, Any]) -> str:
|
| 32 |
dataset_xml = context.get("dataset_xml")
|
| 33 |
readme_text = context.get("readme_text")
|
requirements.txt
CHANGED
|
@@ -1,2 +1,3 @@
|
|
| 1 |
openai>=1.52.0
|
| 2 |
python-dotenv>=1.0.1
|
|
|
|
|
|
| 1 |
openai>=1.52.0
|
| 2 |
python-dotenv>=1.0.1
|
| 3 |
+
dspy
|