|
23 | 23 | const RESOURCE_BACKUP_PATH = `litellm/${RESOURCE_BACKUP_NAME}`; |
24 | 24 | let providers: string[] = []; |
25 | 25 | let selectedProvider: string = ''; |
26 | | - let maxTokens: number | null = null; |
| 26 | + let maxInputTokens: number | null = null; |
| 27 | + let maxOutputTokens: number | null = null; |
27 | 28 |
|
28 | 29 | onMount(() => { |
29 | 30 | const urlParams = new URLSearchParams(window.location.search); |
@@ -127,20 +128,21 @@ We also need to update [${RESOURCE_BACKUP_NAME}](https://github.com/${REPO_FULL_ |
127 | 128 | let loading = true; |
128 | 129 |
|
129 | 130 | $: { |
130 | | - filterResults(query, selectedProvider, maxTokens); |
| 131 | + filterResults(query, selectedProvider, maxInputTokens, maxOutputTokens); |
131 | 132 | } |
132 | 133 |
|
133 | | - function filterResults(query: string, selectedProvider: string, maxTokens: number | null) { |
| 134 | + function filterResults(query: string, selectedProvider: string, maxInputTokens: number | null, maxOutputTokens: number | null) { |
134 | 135 | if (index) { |
135 | 136 | let filteredResults: Item[]; |
136 | 137 |
|
137 | 138 | // Get all items from the index |
138 | 139 | const allItems = index['_docs'] as Item[]; |
139 | 140 |
|
140 | | - // Filter by provider and max_tokens |
| 141 | + // Filter by provider and max_input_tokens and max_output_tokens |
141 | 142 | filteredResults = allItems.filter(item => |
142 | 143 | (!selectedProvider || item.litellm_provider === selectedProvider) && |
143 | | - (maxTokens === null || (item.max_tokens && item.max_tokens >= maxTokens)) |
| 144 | + (maxInputTokens === null || (item.max_input_tokens && item.max_input_tokens >= maxInputTokens)) && |
| 145 | + (maxOutputTokens === null || (item.max_output_tokens && item.max_output_tokens >= maxOutputTokens)) |
144 | 146 | ); |
145 | 147 |
|
146 | 148 | // Then, apply search query if it's not empty |
@@ -225,13 +227,23 @@ We also need to update [${RESOURCE_BACKUP_NAME}](https://github.com/${REPO_FULL_ |
225 | 227 | </div> |
226 | 228 |
|
227 | 229 | <div class="filter-item"> |
228 | | - <label for="maxTokens">max_tokens >=</label> |
| 230 | + <label for="maxInputTokens">max_input_tokens >=</label> |
229 | 231 | <input |
230 | | - id="maxTokens" |
231 | | - bind:value={maxTokens} |
| 232 | + id="maxInputTokens" |
| 233 | + bind:value={maxInputTokens} |
232 | 234 | type="number" |
233 | 235 | min="0" |
234 | | - placeholder="Enter minimum max_tokens" |
| 236 | + placeholder="Enter minimum max_input_tokens" |
| 237 | + /> |
| 238 | + </div> |
| 239 | + <div class="filter-item"> |
| 240 | + <label for="maxOutputTokens">max_output_tokens >=</label> |
| 241 | + <input |
| 242 | + id="maxOutputTokens" |
| 243 | + bind:value={maxOutputTokens} |
| 244 | + type="number" |
| 245 | + min="0" |
| 246 | + placeholder="Enter minimum max_output_tokens" |
235 | 247 | /> |
236 | 248 | </div> |
237 | 249 | </div> |
|
0 commit comments