Skip to content

Commit 50d142f

Browse files
committed
Format files
1 parent bbb1efd commit 50d142f

File tree

5 files changed

+21
-33
lines changed

5 files changed

+21
-33
lines changed

openrouter-api/ask_auto_model.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,13 @@
55

66
api_key = os.getenv("OPENROUTER_API_KEY")
77

8-
headers = {
9-
"Authorization": f"Bearer {api_key}",
10-
"Content-Type": "application/json"
11-
}
8+
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
129
payload = {
1310
"model": "openrouter/auto",
14-
"messages": [{"role": "user", "content": "Say hello in one sentence."}]
11+
"messages": [{"role": "user", "content": "Say hello in one sentence."}],
1512
}
1613
response = requests.post(OPENROUTER_API_URL, headers=headers, json=payload)
1714
data = response.json()
1815

1916
print(f"Model: {data.get('model')}")
20-
print(f"Response: {data['choices'][0]['message']['content']}")
17+
print(f"Response: {data['choices'][0]['message']['content']}")

openrouter-api/ask_specific_model.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,17 @@
55

66
api_key = os.getenv("OPENROUTER_API_KEY")
77

8-
headers = {
9-
"Authorization": f"Bearer {api_key}",
10-
"Content-Type": "application/json"
11-
}
8+
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
129
payload = {
1310
"model": "openai/gpt-3.5-turbo",
14-
"messages": [{"role": "user", "content": "Say hello in one sentence."}]
11+
"messages": [{"role": "user", "content": "Say hello in one sentence."}],
1512
}
1613
response = requests.post(OPENROUTER_API_URL, headers=headers, json=payload)
1714
data = response.json()
1815

19-
if model := data.get('model'):
16+
if model := data.get("model"):
2017
print(f"Model: {model} by {data['provider']}")
2118
print(f"Response: {data['choices'][0]['message']['content']}")
2219
else:
2320
print("No model found in the response.")
24-
print(f"Response: {data}")
21+
print(f"Response: {data}")

openrouter-api/fallback_models.py

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,28 +5,23 @@
55

66
api_key = os.getenv("OPENROUTER_API_KEY")
77

8+
89
def make_request_with_fallback(models_list, messages):
9-
headers = {
10-
"Authorization": f"Bearer {api_key}",
11-
"Content-Type": "application/json"
12-
}
10+
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
1311
payload = {"models": models_list, "messages": messages}
1412

1513
return requests.post(OPENROUTER_API_URL, headers=headers, json=payload)
1614

15+
1716
response = make_request_with_fallback(
18-
models_list=[
19-
"openai/gpt-5",
20-
"openai/gpt-3.5-turbo",
21-
"openai/gpt-3.5-turbo-16k"
22-
],
23-
messages=[{"role": "user", "content": "What is the capital of France?"}]
17+
models_list=["openai/gpt-5", "openai/gpt-3.5-turbo", "openai/gpt-3.5-turbo-16k"],
18+
messages=[{"role": "user", "content": "What is the capital of France?"}],
2419
)
2520

2621
data = response.json()
27-
if model := data.get('model'):
22+
if model := data.get("model"):
2823
print(f"Model: {model} by {data['provider']}")
2924
print(f"Response: {data['choices'][0]['message']['content']}")
3025
else:
3126
print("No model found in the response.")
32-
print(f"Response: {data}")
27+
print(f"Response: {data}")

openrouter-api/get_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,4 @@
1111

1212
models = data.get("data", [])
1313
print(f"Success! Found {len(models)} models via OpenRouter.")
14-
print(f"Examples: {', '.join(m['id'] for m in models[:5])}")
14+
print(f"Examples: {', '.join(m['id'] for m in models[:5])}")

openrouter-api/route_requests.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,9 @@
55

66
api_key = os.getenv("OPENROUTER_API_KEY")
77

8+
89
def make_request(model, messages, provider_config=None):
9-
headers = {
10-
"Authorization": f"Bearer {api_key}",
11-
"Content-Type": "application/json"
12-
}
10+
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
1311
payload = {"model": model, "messages": messages}
1412
if provider_config:
1513
payload["provider"] = provider_config
@@ -18,15 +16,16 @@ def make_request(model, messages, provider_config=None):
1816
response.raise_for_status()
1917
return response.json()
2018

19+
2120
data = make_request(
2221
model="meta-llama/llama-3.1-70b-instruct",
2322
messages=[{"role": "user", "content": "Explain AI in one sentence."}],
24-
provider_config={"sort": "price"}
23+
provider_config={"sort": "price"},
2524
)
2625

27-
if model := data.get('model'):
26+
if model := data.get("model"):
2827
print(f"Model: {model} by {data['provider']}")
2928
print(f"Response: {data['choices'][0]['message']['content']}")
3029
else:
3130
print("No model found in the response.")
32-
print(f"Response: {data}")
31+
print(f"Response: {data}")

0 commit comments

Comments
 (0)