diagnose scorecard display finetuning
Browse files
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
|
| 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
|
| 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
|