Skip to content

Commit f1a7d20

Browse files
authored
Python snippets: Fix some fkn indentation (#1138)
1 parent a8bedca commit f1a7d20

File tree

1 file changed

+39
-28
lines changed

1 file changed

+39
-28
lines changed

packages/tasks/src/snippets/python.ts

Lines changed: 39 additions & 28 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ ${snippetImportInferenceClient(accessToken, provider)}
5151
messages = ${messagesStr}
5252
5353
stream = client.chat.completions.create(
54-
model="${model.id}",
54+
model="${model.id}",
5555
messages=messages,
5656
${configStr},
5757
stream=True
@@ -62,7 +62,8 @@ for chunk in stream:
6262
},
6363
{
6464
client: "openai",
65-
content: `from openai import OpenAI
65+
content: `\
66+
from openai import OpenAI
6667
6768
client = OpenAI(
6869
base_url="${openAIbaseUrl(provider)}",
@@ -79,7 +80,7 @@ stream = client.chat.completions.create(
7980
)
8081
8182
for chunk in stream:
82-
print(chunk.choices[0].delta.content, end="")`,
83+
print(chunk.choices[0].delta.content, end="")`,
8384
},
8485
];
8586
} else {
@@ -101,7 +102,8 @@ print(completion.choices[0].message)`,
101102
},
102103
{
103104
client: "openai",
104-
content: `from openai import OpenAI
105+
content: `\
106+
from openai import OpenAI
105107
106108
client = OpenAI(
107109
base_url="${openAIbaseUrl(provider)}",
@@ -111,7 +113,7 @@ client = OpenAI(
111113
messages = ${messagesStr}
112114
113115
completion = client.chat.completions.create(
114-
model="${model.id}",
116+
model="${model.id}",
115117
messages=messages,
116118
${configStr}
117119
)
@@ -126,7 +128,8 @@ export const snippetZeroShotClassification = (model: ModelDataMinimal): Inferenc
126128
return [
127129
{
128130
client: "requests",
129-
content: `def query(payload):
131+
content: `\
132+
def query(payload):
130133
response = requests.post(API_URL, headers=headers, json=payload)
131134
return response.json()
132135
@@ -142,7 +145,8 @@ export const snippetZeroShotImageClassification = (model: ModelDataMinimal): Inf
142145
return [
143146
{
144147
client: "requests",
145-
content: `def query(data):
148+
content: `\
149+
def query(data):
146150
with open(data["image_path"], "rb") as f:
147151
img = f.read()
148152
payload={
@@ -153,8 +157,8 @@ export const snippetZeroShotImageClassification = (model: ModelDataMinimal): Inf
153157
return response.json()
154158
155159
output = query({
156-
"image_path": ${getModelInputSnippet(model)},
157-
"parameters": {"candidate_labels": ["cat", "dog", "llama"]},
160+
"image_path": ${getModelInputSnippet(model)},
161+
"parameters": {"candidate_labels": ["cat", "dog", "llama"]},
158162
})`,
159163
},
160164
];
@@ -164,7 +168,8 @@ export const snippetBasic = (model: ModelDataMinimal): InferenceSnippet[] => {
164168
return [
165169
{
166170
client: "requests",
167-
content: `def query(payload):
171+
content: `\
172+
def query(payload):
168173
response = requests.post(API_URL, headers=headers, json=payload)
169174
return response.json()
170175
@@ -179,13 +184,14 @@ export const snippetFile = (model: ModelDataMinimal): InferenceSnippet[] => {
179184
return [
180185
{
181186
client: "requests",
182-
content: `def query(filename):
183-
with open(filename, "rb") as f:
184-
data = f.read()
185-
response = requests.post(API_URL, headers=headers, data=data)
186-
return response.json()
187-
188-
output = query(${getModelInputSnippet(model)})`,
187+
content: `\
188+
def query(filename):
189+
with open(filename, "rb") as f:
190+
data = f.read()
191+
response = requests.post(API_URL, headers=headers, data=data)
192+
return response.json()
193+
194+
output = query(${getModelInputSnippet(model)})`,
189195
},
190196
];
191197
};
@@ -253,12 +259,14 @@ export const snippetTabular = (model: ModelDataMinimal): InferenceSnippet[] => {
253259
return [
254260
{
255261
client: "requests",
256-
content: `def query(payload):
257-
response = requests.post(API_URL, headers=headers, json=payload)
258-
return response.content
259-
response = query({
260-
"inputs": {"data": ${getModelInputSnippet(model)}},
261-
})`,
262+
content: `\
263+
def query(payload):
264+
response = requests.post(API_URL, headers=headers, json=payload)
265+
return response.content
266+
267+
response = query({
268+
"inputs": {"data": ${getModelInputSnippet(model)}},
269+
})`,
262270
},
263271
];
264272
};
@@ -271,7 +279,8 @@ export const snippetTextToAudio = (model: ModelDataMinimal): InferenceSnippet[]
271279
return [
272280
{
273281
client: "requests",
274-
content: `def query(payload):
282+
content: `\
283+
def query(payload):
275284
response = requests.post(API_URL, headers=headers, json=payload)
276285
return response.content
277286
@@ -287,7 +296,8 @@ Audio(audio_bytes)`,
287296
return [
288297
{
289298
client: "requests",
290-
content: `def query(payload):
299+
content: `\
300+
def query(payload):
291301
response = requests.post(API_URL, headers=headers, json=payload)
292302
return response.json()
293303
@@ -306,9 +316,10 @@ export const snippetDocumentQuestionAnswering = (model: ModelDataMinimal): Infer
306316
return [
307317
{
308318
client: "requests",
309-
content: `def query(payload):
310-
with open(payload["image"], "rb") as f:
311-
img = f.read()
319+
content: `\
320+
def query(payload):
321+
with open(payload["image"], "rb") as f:
322+
img = f.read()
312323
payload["image"] = base64.b64encode(img).decode("utf-8")
313324
response = requests.post(API_URL, headers=headers, json=payload)
314325
return response.json()

0 commit comments

Comments
 (0)