Chris Addis commited on
Commit
46f2212
·
1 Parent(s): 3a3681c

add photograph

Browse files
Files changed (1) hide show
  1. 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 request parameters including messages
143
- request_data = {
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
- request_data["messages"].append({
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}", "detail": detail},
170
  })
171
 
172
- request_data["messages"].append({
173
  "role": "user",
174
- "content": content
 
 
 
175
  })
176
 
177
- # Make the API call with the complete request data
178
- response = self.client.chat.completions.create(**request_data)
 
 
 
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