aadisawant2912 commited on
Commit
fd854d2
·
verified ·
1 Parent(s): 061ad54

Update tools_v2.py

Browse files
Files changed (1) hide show
  1. tools_v2.py +4 -4
tools_v2.py CHANGED
@@ -529,13 +529,13 @@ def label_clusters_council_of_3(batch_size: int = 5) -> str:
529
  **cluster,
530
  "label": final,
531
  "llm_vote_1_MISTRAL": votes[0],
532
- "llm_vote_2_OPENAI": votes[1],
533
  "llm_vote_3_GROQ": votes[2],
534
  "confidence_1": persona_results.get("MISTRAL", {}).get(cid, {}).get("confidence", ""),
535
- "confidence_2": persona_results.get("OPENAI", {}).get(cid, {}).get("confidence", ""),
536
  "confidence_3": persona_results.get("GROQ", {}).get(cid, {}).get("confidence", ""),
537
  "reasoning_1": persona_results.get("MISTRAL", {}).get(cid, {}).get("reasoning", ""),
538
- "reasoning_2": persona_results.get("OPENAI", {}).get(cid, {}).get("reasoning", ""),
539
  "reasoning_3": persona_results.get("GROQ", {}).get(cid, {}).get("reasoning", ""),
540
  "vote_agreement": agreement,
541
  }
@@ -554,7 +554,7 @@ def label_clusters_council_of_3(batch_size: int = 5) -> str:
554
  "final_label": c["label"],
555
  "vote_agreement": c["vote_agreement"],
556
  "llm1_MISTRAL_label": c["llm_vote_1_MISTRAL"],
557
- "llm2_OPENAI_label": c["llm_vote_2_OPENAI"],
558
  "llm3_GROQ_label": c["llm_vote_3_GROQ"],
559
  "llm1_confidence": c["confidence_1"],
560
  "llm2_confidence": c["confidence_2"],
 
529
  **cluster,
530
  "label": final,
531
  "llm_vote_1_MISTRAL": votes[0],
532
+ "llm_vote_2_GEMINI": votes[1],
533
  "llm_vote_3_GROQ": votes[2],
534
  "confidence_1": persona_results.get("MISTRAL", {}).get(cid, {}).get("confidence", ""),
535
+ "confidence_2": persona_results.get("GEMINI", {}).get(cid, {}).get("confidence", ""),
536
  "confidence_3": persona_results.get("GROQ", {}).get(cid, {}).get("confidence", ""),
537
  "reasoning_1": persona_results.get("MISTRAL", {}).get(cid, {}).get("reasoning", ""),
538
+ "reasoning_2": persona_results.get("GEMINI", {}).get(cid, {}).get("reasoning", ""),
539
  "reasoning_3": persona_results.get("GROQ", {}).get(cid, {}).get("reasoning", ""),
540
  "vote_agreement": agreement,
541
  }
 
554
  "final_label": c["label"],
555
  "vote_agreement": c["vote_agreement"],
556
  "llm1_MISTRAL_label": c["llm_vote_1_MISTRAL"],
557
+ "llm2_GEMINI_label": c["llm_vote_2_GEMINI"],
558
  "llm3_GROQ_label": c["llm_vote_3_GROQ"],
559
  "llm1_confidence": c["confidence_1"],
560
  "llm2_confidence": c["confidence_2"],