NewLaptop_ commited on
Commit
90df45b
·
2 Parent(s): 79e5346 9d1b729

Merge branch 'main' of https://github.com/stefanches7/AI-assisted-Neuroimaging-harmonization

Browse files
Files changed (2) hide show
  1. agent.py +9 -2
  2. requirements.txt +1 -0
agent.py CHANGED
@@ -13,14 +13,21 @@ class BIDSifierAgent:
13
  load_dotenv()
14
 
15
  if provider=="openai":
16
- lm = dspy.LM(f"{provider}/{model}", api_key=os.getenv("OPENAI_API_KEY"))
 
 
 
 
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