From 0845e2763705928df2463a3f42a2ea2ba6e38dc0 Mon Sep 17 00:00:00 2001
From: Shengye Wan <simonwan@meta.com>
Date: Wed, 6 Dec 2023 18:39:11 -0800
Subject: [PATCH] Fix the TOGETHER class

Summary:
Together API changes the field `model_name` to `model` now:
```
pydantic.v1.error_wrappers.ValidationError: 2 validation errors for Together
model
  field required (type=value_error.missing)
model_name
  extra fields not permitted (type=value_error.extra)
```
Hence make the update.

Reviewed By: spencerwmeta

Differential Revision: D51925200

fbshipit-source-id: fb285d0ace46c71d572ce1e401a14bf0c55bcedc
---
 CybersecurityBenchmarks/benchmark/llm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/CybersecurityBenchmarks/benchmark/llm.py b/CybersecurityBenchmarks/benchmark/llm.py
index 9b7dd6954..e03dc5fba 100644
--- a/CybersecurityBenchmarks/benchmark/llm.py
+++ b/CybersecurityBenchmarks/benchmark/llm.py
@@ -177,7 +177,7 @@ class TOGETHER(LLM):
     @override
     def query(self, prompt: str) -> str:
         llm = Together(
-            model_name=self.model,
+            model=self.model,
             temperature=0.75,
             top_p=1,
             max_tokens=MAX_TOKENS,