BtB-ExpC commited on
Commit
bcccfb8
·
1 Parent(s): 40aa0fe

diagnose scorecard display finetuning

Browse files
Files changed (1) hide show
  1. chains/diagnoser_chain.py +3 -3
chains/diagnoser_chain.py CHANGED
@@ -33,7 +33,7 @@ class DiagnoserChain(BaseModel):
33
  messages = prompt.to_messages()
34
  diagnosis_response = await self.llm_diagnose.ainvoke(messages)
35
  content = diagnosis_response.content if hasattr(diagnosis_response, "content") else diagnosis_response
36
- return f"[DIAGNOSIS {idx}]{content}"
37
 
38
  # Launch all diagnosis tasks concurrently.
39
  tasks = [
@@ -43,7 +43,7 @@ class DiagnoserChain(BaseModel):
43
  diagnoses = await asyncio.gather(*tasks)
44
 
45
  # Step 3: Combine the outputs from each prompt.
46
- combined_diagnosis = "\n\n---\n".join(diagnoses)
47
 
48
  # Step 4: Generate scorecard
49
  prompt = await self.template_diagnose_scorecard.aformat_prompt(combined_diagnosis=combined_diagnosis)
@@ -51,7 +51,7 @@ class DiagnoserChain(BaseModel):
51
  scorecard_response = await self.llm_diagnose.ainvoke(scorecard_messages)
52
  scorecard = scorecard_response.content if hasattr(scorecard_response, "content") else scorecard_response
53
 
54
- return scorecard + "\n" + combined_diagnosis
55
 
56
  class Config:
57
  arbitrary_types_allowed = True
 
33
  messages = prompt.to_messages()
34
  diagnosis_response = await self.llm_diagnose.ainvoke(messages)
35
  content = diagnosis_response.content if hasattr(diagnosis_response, "content") else diagnosis_response
36
+ return f"--- [DIAGNOSIS {idx}] --- \n{content}"
37
 
38
  # Launch all diagnosis tasks concurrently.
39
  tasks = [
 
43
  diagnoses = await asyncio.gather(*tasks)
44
 
45
  # Step 3: Combine the outputs from each prompt.
46
+ combined_diagnosis = "\n\n".join(diagnoses)
47
 
48
  # Step 4: Generate scorecard
49
  prompt = await self.template_diagnose_scorecard.aformat_prompt(combined_diagnosis=combined_diagnosis)
 
51
  scorecard_response = await self.llm_diagnose.ainvoke(scorecard_messages)
52
  scorecard = scorecard_response.content if hasattr(scorecard_response, "content") else scorecard_response
53
 
54
+ return combined_diagnosis + "\n\n" + scorecard
55
 
56
  class Config:
57
  arbitrary_types_allowed = True