Chris Addis commited on
Commit ·
46f2212
1
Parent(s): 3a3681c
add photograph
Browse files- library/utils_model.py +14 -15
library/utils_model.py
CHANGED
|
@@ -139,19 +139,12 @@ class OpenRouterAPI:
|
|
| 139 |
# Getting the Base64 string with optional resizing
|
| 140 |
base64_image = encode_image(image_path, max_size=max_image_size)
|
| 141 |
|
| 142 |
-
# Prepare
|
| 143 |
-
|
| 144 |
-
"model": model,
|
| 145 |
-
"messages": [],
|
| 146 |
-
"temperature": temperature,
|
| 147 |
-
'provider': {
|
| 148 |
-
'data_collection': 'deny'
|
| 149 |
-
}
|
| 150 |
-
}
|
| 151 |
|
| 152 |
# Add system message if prompt_dev is provided
|
| 153 |
if prompt_dev:
|
| 154 |
-
|
| 155 |
"role": "system",
|
| 156 |
"content": prompt_dev
|
| 157 |
})
|
|
@@ -166,16 +159,22 @@ class OpenRouterAPI:
|
|
| 166 |
|
| 167 |
content.append({
|
| 168 |
"type": "image_url",
|
| 169 |
-
"image_url": {"url": f"data:image/jpeg;base64,{base64_image}"
|
| 170 |
})
|
| 171 |
|
| 172 |
-
|
| 173 |
"role": "user",
|
| 174 |
-
"content": content
|
|
|
|
|
|
|
|
|
|
| 175 |
})
|
| 176 |
|
| 177 |
-
|
| 178 |
-
|
|
|
|
|
|
|
|
|
|
| 179 |
|
| 180 |
return response.choices[0].message.content
|
| 181 |
|
|
|
|
| 139 |
# Getting the Base64 string with optional resizing
|
| 140 |
base64_image = encode_image(image_path, max_size=max_image_size)
|
| 141 |
|
| 142 |
+
# Prepare messages based on OpenRouter's format
|
| 143 |
+
messages = []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 144 |
|
| 145 |
# Add system message if prompt_dev is provided
|
| 146 |
if prompt_dev:
|
| 147 |
+
messages.append({
|
| 148 |
"role": "system",
|
| 149 |
"content": prompt_dev
|
| 150 |
})
|
|
|
|
| 159 |
|
| 160 |
content.append({
|
| 161 |
"type": "image_url",
|
| 162 |
+
"image_url": {"url": f"data:image/jpeg;base64,{base64_image}"},
|
| 163 |
})
|
| 164 |
|
| 165 |
+
messages.append({
|
| 166 |
"role": "user",
|
| 167 |
+
"content": content,
|
| 168 |
+
"provider": {
|
| 169 |
+
"data_collection": "deny"
|
| 170 |
+
}
|
| 171 |
})
|
| 172 |
|
| 173 |
+
response = self.client.chat.completions.create(
|
| 174 |
+
model=model,
|
| 175 |
+
messages=messages,
|
| 176 |
+
temperature=temperature
|
| 177 |
+
)
|
| 178 |
|
| 179 |
return response.choices[0].message.content
|
| 180 |
|