Datawithsarah commited on
Commit
e949b7f
·
1 Parent(s): 9e54048

correct Claude v1 API usage

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -29,27 +29,26 @@ class ClaudeServerModel:
29
  self.model_id = model_id
30
  self.temperature = temperature
31
 
32
- def complete(self, prompt: str, stop_sequences: list[str] = None) -> str:
33
  headers = {
34
  "x-api-key": self.api_key,
35
  "anthropic-version": "2023-06-01",
36
  "content-type": "application/json"
37
  }
 
38
  body = {
39
  "model": self.model_id,
40
  "max_tokens": 1024,
41
  "temperature": self.temperature,
42
- "messages": [{"role": "user", "content": prompt}]
43
  }
44
- if stop_sequences:
45
- body["stop_sequences"] = stop_sequences
46
 
47
- response = requests.post("https://api.anthropic.com/v1/messages", headers=headers, json=body)
48
  response.raise_for_status()
49
- return response.json()["content"][0]["text"].strip()
50
 
51
- def __call__(self, prompt: str, stop_sequences: list[str] = None) -> str:
52
- return self.complete(prompt, stop_sequences=stop_sequences)
53
 
54
  # --- Constants ---
55
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
29
  self.model_id = model_id
30
  self.temperature = temperature
31
 
32
+ def complete(self, prompt: str) -> str:
33
  headers = {
34
  "x-api-key": self.api_key,
35
  "anthropic-version": "2023-06-01",
36
  "content-type": "application/json"
37
  }
38
+
39
  body = {
40
  "model": self.model_id,
41
  "max_tokens": 1024,
42
  "temperature": self.temperature,
43
+ "prompt": f"\n\nHuman: {prompt}\n\nAssistant:"
44
  }
 
 
45
 
46
+ response = requests.post("https://api.anthropic.com/v1/complete", headers=headers, json=body)
47
  response.raise_for_status()
48
+ return response.json()["completion"].strip()
49
 
50
+ def __call__(self, prompt: str) -> str:
51
+ return self.complete(prompt)
52
 
53
  # --- Constants ---
54
  DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"