diff --git a/.github/mcp-config.json b/.github/mcp-config.json deleted file mode 100644 index 0967ef424b..0000000000 --- a/.github/mcp-config.json +++ /dev/null @@ -1 +0,0 @@ -{} diff --git a/.gitignore b/.gitignore index 4df1885fc9..89029b28ee 100644 --- a/.gitignore +++ b/.gitignore @@ -338,6 +338,8 @@ values.yaml /security-report.md /security-reports/ /test-results/ +tests/test-results/ +tests/security/security-reports/ config/test-results/ config/playwright-report/ *_output.txt @@ -501,6 +503,8 @@ infra/sinker/run_*.json HANDOFF_training_dataset.md Modern Multi-Agent Coding Architectures.pdf *.bak +*.log.old +*.log.v2 NF_PIPELINE_HANDOFF.md thread-5302e593-transcript.md .jscpd-report/ diff --git a/apps/web/src/components/ai/SyntheticTherapyDemo.tsx.disabled b/apps/web/src/components/ai/SyntheticTherapyDemo.tsx.disabled deleted file mode 100644 index a4f26e6e0e..0000000000 --- a/apps/web/src/components/ai/SyntheticTherapyDemo.tsx.disabled +++ /dev/null @@ -1,630 +0,0 @@ -import { useState } from 'react' -import { Button } from '@/components/ui/button' -import { - Card, - CardHeader, - CardTitle, - CardDescription, - CardContent, - CardFooter, -} from '@/components/ui/card' -import { Tabs, TabsContent, TabsList, TabsTrigger } from '@/components/ui/tabs' -import { - Select, - SelectContent, - SelectItem, - SelectTrigger, - SelectValue, -} from '@/components/ui/select' -import { Slider } from '@/components/ui/slider' -import { Switch } from '@/components/ui/switch' -import { Label } from '@/components/ui/label' -import { Badge } from '@/components/ui/badge' -import { - Tooltip, - TooltipContent, - TooltipProvider, - TooltipTrigger, -} from '@/components/ui/tooltip' -import { - InfoIcon, - BrainIcon, - MessageSquareIcon, - RefreshCwIcon, - DownloadIcon, -} from 'lucide-react' -import { DisorderCategory } from '@/lib/ai/mental-arena/types' - -interface SyntheticConversation { - patientText: string - therapistText: string - encodedSymptoms: Array<{ - name: string - severity: number - duration: string - manifestations: string[] - cognitions: string[] - }> - decodedSymptoms: string[] - sessionSummary?: string - accuracyScore?: number -} - -/** - * Component for demonstrating synthetic therapy conversation generation - */ -export default function SyntheticTherapyDemo() { - const [loading, setLoading] = useState(false) - const [conversations, setConversations] = useState( - [], - ) - const [selectedConversationIndex, setSelectedConversationIndex] = useState(0) - const [config, setConfig] = useState({ - numSessions: 1, - maxTurns: 3, - disorders: [DisorderCategory.Anxiety, DisorderCategory.Depression], - usePythonBridge: false, - model: 'gpt-3.5-turbo', - }) - - // Handle generating therapy conversations - const handleGenerateConversations = async () => { - setLoading(true) - try { - // Call our new psychology scenario generation API - const response = await fetch('/api/psychology/generate-scenario', { - method: 'POST', - headers: { - 'Content-Type': 'application/json', - }, - body: JSON.stringify({ - type: 'therapy_conversation', - difficulty: params.disorders.length > 1 ? 'moderate' : 'basic', - clientProfile: { - disorders: params.disorders, - sessionNumber: Math.floor(Math.random() * 10) + 1, - presenting_concerns: params.disorders.map(d => `Primary concern related to ${d}`), - demographics: { - age: Math.floor(Math.random() * 40) + 20, - gender: Math.random() > 0.5 ? 'female' : 'male' - } - }, - therapeuticFramework: 'CBT', // Default to CBT - options: { - includeSymptoms: true, - includeAnalysis: true, - realistic: true, - evidenceBased: true - } - }) - }) - - if (!response.ok) { - throw new Error(`API request failed: ${response.status}`) - } - - const scenarioResult = await response.json() - - // Transform API response to match our conversation format - const apiConversations: SyntheticConversation[] = [{ - patientText: scenarioResult.scenario.clientStatement, - therapistText: scenarioResult.scenario.therapistResponse, - encodedSymptoms: scenarioResult.scenario.symptoms.map((symptom: any) => ({ - name: symptom.name, - severity: symptom.severity / 10, // Convert 1-10 to 0-1 - duration: symptom.duration, - manifestations: symptom.indicators, - cognitions: symptom.cognitivePatterns || [] - })), - decodedSymptoms: scenarioResult.analysis.identifiedSymptoms.map((symptom: any) => symptom.name), - sessionSummary: scenarioResult.analysis.clinicalSummary, - accuracyScore: scenarioResult.analysis.accuracyScore || 0.85 - }] - - setConversations(apiConversations) - - } catch (error) { - console.error('Failed to generate conversations:', error) - - // Fallback to mock data on API failure - const mockConversations: SyntheticConversation[] = [ - { - patientText: - "I've been struggling with excessive worry for about 6 months now. I have difficulty sleeping and I'm constantly catastrophizing. I'm also experiencing restlessness and fatigue. Fidgeting has been particularly difficult to deal with. I'm not sure what to do anymore. I've tried to manage on my own, but it's getting harder. Can you help me understand what's happening and what I might do about it?", - therapistText: - "I hear that you've been dealing with excessive worry for quite some time, and it's been affecting your sleep and causing physical symptoms like restlessness and fatigue. That sounds really challenging. When you mention catastrophizing, could you share an example of the kinds of thoughts you have when you're worried?", - encodedSymptoms: [ - { - name: 'excessive worry', - severity: 0.7, - duration: '6 months', - manifestations: [ - 'difficulty sleeping', - 'restlessness', - 'physical tension', - 'avoidance of anxiety-provoking situations', - ], - cognitions: [ - 'catastrophizing', - 'overestimation of threat', - 'intolerance of uncertainty', - ], - }, - { - name: 'restlessness', - severity: 0.6, - duration: '3 months', - manifestations: ['fidgeting', 'unable to sit still', 'pacing'], - cognitions: ['feeling on edge', 'anticipating danger'], - }, - { - name: 'fatigue', - severity: 0.5, - duration: '2 months', - manifestations: [ - 'decreased energy', - 'difficulty completing tasks', - 'requiring more rest than usual', - ], - cognitions: ['feeling overwhelmed', 'diminished self-efficacy'], - }, - ], - decodedSymptoms: ['anxiety', 'insomnia', 'fatigue'], - sessionSummary: - "Session Summary:\n\nPatient presented with excessive worry, restlessness, fatigue.\nTherapist identified: anxiety, insomnia, fatigue.\n\nSymptom detection accuracy: 67%\n\nThe conversation covered the patient's experiences with excessive worry, fatigue.\nThe therapist may have missed: restlessness.\n\nThis simulated interaction demonstrates the importance of thorough assessment and active listening in the therapeutic relationship.", - accuracyScore: 0.67, - }, - ] - - setConversations(mockConversations) - setSelectedConversationIndex(0) - } finally { - setLoading(false) - } - } - - const selectedConversation = conversations[selectedConversationIndex] || null - - return ( -
-
-

- Synthetic Therapy Conversation Generator -

- - - - - - -

- This demo uses the MentalArena framework to generate synthetic - therapy conversations between patients and therapists. The - system encodes symptoms into a patient profile and then measures - how accurately a therapist model identifies those symptoms. -

-
-
-
-
- -
- - - Configuration - - Adjust settings for conversation generation - - - -
- - - setConfig({ ...config, numSessions: value[0] ?? 1 }) - } - /> - -
- {config.numSessions} -
-
- -
- - - setConfig({ ...config, maxTurns: value[0] ?? 3 }) - } - /> - -
- {config.maxTurns} -
-
- -
- - -
- -
- -
- {(Object.values(DisorderCategory) as string[]) - .slice(0, 5) - .map((disorder) => ( - { - const disorders = config.disorders.includes( - disorder as DisorderCategory, - ) - ? config.disorders.filter((d) => d !== disorder) - : [...config.disorders, disorder as DisorderCategory] - setConfig({ ...config, disorders }) - }} - > - {disorder} - - ))} -
-
- -
- - setConfig({ ...config, usePythonBridge: checked }) - } - /> - - -
-
- - - -
- -
- {conversations.length > 0 && selectedConversation ? ( - <> - - - Conversation - Symptoms - Summary - - - - - - - Synthetic Therapy Conversation - - Session {selectedConversationIndex + 1}/ - {conversations.length} - - - - -
-
-
- P -
-
-
Patient
-
- {selectedConversation.patientText} -
-
-
-
- -
-
-
- T -
-
-
Therapist
-
- {selectedConversation.therapistText} -
-
-
-
-
-
-
- - -
- - - Encoded Symptoms (Patient) - - Symptoms encoded into the patient profile - - - - {selectedConversation.encodedSymptoms.map( - (symptom) => ( -
-
-

{symptom.name}

- - {symptom.duration} - -
-
-
- Severity:{' '} - {(symptom.severity * 100).toFixed(0)}% -
-
-
-
-
-
-
- Manifestations: -
-
- {symptom.manifestations.map( - (manifestation) => ( - - {manifestation} - - ), - )} -
-
-
-
- Cognitions: -
-
- {symptom.cognitions.map((cognition) => ( - - {cognition} - - ))} -
-
-
- ), - )} - - - - - - Decoded Symptoms (Therapist) - - Symptoms identified by the therapist model - - - -
-
Identified Symptoms
-
- {selectedConversation.decodedSymptoms.map( - (symptom) => { - const isCorrect = - selectedConversation.encodedSymptoms.some( - (s) => - s.name.includes(symptom) || - symptom.includes(s.name), - ) - return ( - - {symptom} - - ) - }, - )} -
-
- -
-
- Correctly Identified -
-
- {selectedConversation.encodedSymptoms - .filter((encoded) => - selectedConversation.decodedSymptoms.some( - (decoded) => - decoded.includes(encoded.name) || - encoded.name.includes(decoded), - ), - ) - .map((symptom) => ( - - {symptom.name} - - ))} -
-
- -
-
Missed by Therapist
-
- {selectedConversation.encodedSymptoms - .filter( - (encoded) => - !selectedConversation.decodedSymptoms.some( - (decoded) => - decoded.includes(encoded.name) || - encoded.name.includes(decoded), - ), - ) - .map((symptom) => ( - - {symptom.name} - - ))} -
-
- -
-
- Incorrectly Identified -
-
- {selectedConversation.decodedSymptoms - .filter( - (decoded) => - !selectedConversation.encodedSymptoms.some( - (encoded) => - encoded.name.includes(decoded) || - decoded.includes(encoded.name), - ), - ) - .map((symptom) => ( - - {symptom} - - ))} -
-
-
-
-
- - - - - - Session Summary & Analysis - - AI-generated summary and accuracy score - - - -
- {selectedConversation.sessionSummary} -
-
- = 0.7 - ? 'default' - : 'destructive' - } - > - {( - (selectedConversation.accuracyScore || 0) * 100 - ).toFixed(0)} - % Accuracy - -
-
- - - -
-
- - - ) : ( - - - No Conversations Generated - - Click the button to generate synthetic conversations. - - - - - - - )} -
-
-
- ) -} diff --git a/apps/web/src/components/analytics/TableWidget.tsx.bak b/apps/web/src/components/analytics/TableWidget.tsx.bak deleted file mode 100644 index df02e6940e..0000000000 --- a/apps/web/src/components/analytics/TableWidget.tsx.bak +++ /dev/null @@ -1,317 +0,0 @@ -import React, { useState, useEffect, useCallback, useMemo } from 'react'; -import { DashboardWidget } from './DashboardWidget'; -import type { DashboardWidgetProps as WidgetProps } from './DashboardWidget'; -import { - TableBody, - TableCell, - TableHead, - TableHeader, - TableRow, -} from '@/components/ui/table'; -import { Input } from '@/components/ui/input'; -import { Button } from '@/components/ui/button'; -import { Download, Search, ArrowUp, ArrowDown } from 'lucide-react'; -// TableState type is not currently used in this component - -// Define a generic type for table row data -export type TableRowData = Record & { - id?: string | number - [key: string]: unknown -} - -export interface Column { - key: string - label: string - render?: (value: unknown, row: TableRowData) => React.ReactNode - sortable?: boolean - filterable?: boolean -} - -export interface TableWidgetProps - extends Omit< - WidgetProps, - 'children' | 'title' | 'description' | 'isLoading' - > { - /** The title of the table widget */ - title: string - - /** Optional description for the table */ - description?: string - - /** Array of column definitions */ - columns: Column[] - - /** Array of row data */ - data: TableRowData[] - - /** Whether the table is in a loading state */ - isLoading?: boolean - - /** Optional class name for the root element */ - className?: string - - /** Refresh interval in milliseconds */ - refreshInterval?: number - - /** Whether to show the search input */ - enableSearch?: boolean - - /** Whether to show the export button */ - enableExport?: boolean - - /** Function to fetch data asynchronously */ - fetchData?: () => Promise - - /** Pagination configuration */ - pagination?: { - pageSize: number - initialPage?: number - } -} - -export function TableWidget({ - title, - description, - columns, - data: initialData, - isLoading: initialLoading = false, - className = '', - refreshInterval, - enableSearch = true, - enableExport = true, - fetchData, - pagination, - ...props -}: TableWidgetProps): JSX.Element { - const [data, setData] = useState(initialData) - const [isLoading, setIsLoading] = useState(initialLoading) - const [error, setError] = useState(null) - const [searchTerm, setSearchTerm] = useState('') - const [sortConfig, setSortConfig] = useState<{ - key: string - direction: 'asc' | 'desc' - } | null>(null) - // Handle data fetching if fetchData is provided - useEffect(() => { - if (fetchData === undefined) { - return - } - - const loadData = async () => { - try { - setIsLoading(true) - const result = await fetchData() - setData(result) - } catch (error) { - console.error('Error fetching table data:', error) - } finally { - setIsLoading(false) - } - } - - loadData() - - // Set up refresh interval if provided - let intervalId: NodeJS.Timeout | null = null - if (refreshInterval && refreshInterval > 0) { - intervalId = setInterval(loadData, refreshInterval) - } - - return () => { - if (intervalId) { - clearInterval(intervalId) - } - } - }, [fetchData, refreshInterval, setIsLoading, setData]) - - // Handle sorting - const handleSort = useCallback((key: string) => { - setSortConfig((prevConfig) => { - // If the same key is clicked, toggle the direction - if (prevConfig?.key === key) { - return { - key, - direction: prevConfig.direction === 'asc' ? 'desc' : 'asc', - } - } - // Default to ascending for a new key - return { key, direction: 'asc' } - }) - }, []) - - // Apply sorting and filtering - const filteredAndSortedData = useMemo(() => { - let result = [...data] - - // Apply search filter - if (searchTerm) { - const searchLower = searchTerm.toLowerCase() - result = result.filter((row) => - columns.some((column) => { - const value = String(row[column.key] || '').toLowerCase() - return value.includes(searchLower) - }), - ) - } - - // Apply sorting - if (sortConfig) { - const { key, direction } = sortConfig - result.sort((a, b) => { - const aValue = a[key] - const bValue = b[key] - - if (aValue === bValue) { - return 0; - } - if (aValue == null) { - return direction === 'asc' ? -1 : 1; - } - if (bValue == null) { - return direction === 'asc' ? 1 : -1; - } - - const comparison = String(aValue).localeCompare(String(bValue)) - return direction === 'asc' ? comparison : -comparison - }) - } - - // Pagination is not currently implemented - // The pagination object is intentionally unused for now - void pagination; - - return result - }, [data, searchTerm, sortConfig, columns, pagination]) - // Handle export - const handleExport = useCallback(() => { - try { - // Simple CSV export implementation - const headers = columns.map((col) => `"${col.label}"`).join(',') - const rows = data - .map((row) => - columns - .map((col) => { - const value = row[col.key] - // Escape quotes and wrap in quotes - const escaped = String(value || '').replace(/"/g, '""') - return `"${escaped}"` - }) - .join(','), - ) - .join('\n') - - const csvContent = `${headers}\n${rows}` - const blob = new Blob([csvContent], { type: 'text/csv;charset=utf-8;' }) - const url = URL.createObjectURL(blob) - const link = document.createElement('a') - link.href = url - link.download = `${title.toLowerCase().replace(/\s+/g, '-')}-export-${new Date().toISOString().slice(0, 10)}.csv` - document.body.appendChild(link) - link.click() - document.body.removeChild(link) - URL.revokeObjectURL(url) - } catch (error) { - console.error('Error exporting data:', error) - } - }, [data, columns, title]) - - return ( - -
- {/* Search and export controls */} -
- {enableSearch && ( -
- - setSearchTerm(e.target.value)} - /> -
- )} - - {enableExport && ( - - )} -
- - {/* Table */} -
- {/* Using a basic table structure instead of the Table component for now */} - - - - {columns.map((column) => ( - - {column.sortable ? ( - - ) : ( - column.label - )} - - ))} - - - - {data.length === 0 ? ( - - - No data available - - - ) : ( - data.map((row, rowIndex) => ( - - {columns.map((column) => ( - - {column.render - ? column.render(row[column.key], row) - : String(row[column.key] || '')} - - ))} - - )) - )} - -
-
- - {/* Pagination would go here */} -
-
- ) -} diff --git a/apps/web/src/hooks/__tests__/fix-example.txt b/apps/web/src/hooks/__tests__/fix-example.txt deleted file mode 100644 index 5804b6665e..0000000000 --- a/apps/web/src/hooks/__tests__/fix-example.txt +++ /dev/null @@ -1,70 +0,0 @@ -/** - * Example fix for useNotificationPreferences test - * - * The error is: "expected [object Object] to equal ObjectContaining" - * This usually happens when you're comparing objects incorrectly with expect.objectContaining - * - * Original test might look like: - */ - -// Original failing test -test('falls back to default preferences on load error', async () => { - // Mock a failed API call - mockApi.getUserNotificationPreferences.mockRejectedValue(new Error('API Error')); - - const { result } = renderHook(() => useNotificationPreferences()); - - // Wait for the hook to handle the error - await waitFor(() => { - // This assertion is failing because objectContaining isn't being used correctly - expect(result.current.preferences).toEqual(expect.objectContaining(DEFAULT_PREFERENCES)); - }); -}); - -/** - * The fix has two possible solutions: - * - * 1. Use expect.objectContaining() properly: - */ - -// Fixed test - solution 1 -test('falls back to default preferences on load error', async () => { - mockApi.getUserNotificationPreferences.mockRejectedValue(new Error('API Error')); - - const { result } = renderHook(() => useNotificationPreferences()); - - await waitFor(() => { - // Now we're using objectContaining as a matcher function - expect(result.current.preferences).toEqual(expect.objectContaining(DEFAULT_PREFERENCES)); - }); -}); - -/** - * 2. Or use direct equality if the objects should be exactly the same: - */ - -// Fixed test - solution 2 -test('falls back to default preferences on load error', async () => { - mockApi.getUserNotificationPreferences.mockRejectedValue(new Error('API Error')); - - const { result } = renderHook(() => useNotificationPreferences()); - - await waitFor(() => { - // Direct equality comparison if the objects should be exactly the same - expect(result.current.preferences).toEqual(DEFAULT_PREFERENCES); - }); -}); - -/** - * Note: The problem could also be in the implementation of objectContaining itself. - * Check vitest.setup.ts to make sure objectContaining is properly implemented: - * - * const originalObjectContaining = expect.objectContaining; - * expect.objectContaining = (obj) => { - * const matcher = originalObjectContaining(obj); - * matcher.toString = () => `ObjectContaining(${JSON.stringify(obj)})`; - * return matcher; - * }; - * - * This ensures that the matcher has a proper string representation. - */ diff --git a/coverage-audit-report.md b/coverage-audit-report.md deleted file mode 100644 index 6e860cbcbb..0000000000 --- a/coverage-audit-report.md +++ /dev/null @@ -1,10 +0,0 @@ -# Coverage Audit Report - -- Python: `937 passed, 23 skipped` with `fail_under = 20` in `pyproject.toml` - (baseline measured ~21% for `src` + `ai`). -- TypeScript governance/audit/fhe/security suites pass via `VITEST_BUCKET=1` - scoped execution; full-suite baseline thresholds remain 55 lines / 55 - statements / 55 functions / 45 branches in `vitest.config.ts`. -- HIPAA and security suites exist per the test coverage plan. - -Generated 2026-10-21. diff --git a/scripts/training/training.log.old b/scripts/training/training.log.old deleted file mode 100644 index 5f977dd0eb..0000000000 --- a/scripts/training/training.log.old +++ /dev/null @@ -1,895 +0,0 @@ -/home/vivi/pixelated/.venv/lib/python3.13/site-packages/torch/cuda/__init__.py:64: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you. - import pynvml # type: ignore[import] -Loading dataset... -Loaded 19208 total examples. -Deduped dataset written to /tmp/tmppvtyyoba.jsonl -=== Training attempt #1 === -weave: Logged in as Weights & Biases user: wutang. -weave: View Weave data at https://wandb.ai/wutang/wayfarer-ab-test/weave -Logged in as Weights & Biases user: wutang. -View Weave data at https://wandb.ai/wutang/wayfarer-ab-test/weave -Loading model... -W&B step: 0, checkpoint: 0, resuming from: 0 -Starting SFT Warmup (50 steps)... -wandb: [wandb.login()] Using explicit session credentials for https://api.wandb.ai. -wandb: Currently logged in as: wutang to https://api.wandb.ai. Use `wandb login --relogin` to force relogin -wandb: Tracking run with wandb version 0.28.1 -wandb: Run data is saved locally in /data/vivi/pixelated/scripts/training/wandb/run-20260723_183309-qwen3-30b-serverless-rl-long-20260723-1832 -wandb: Run `wandb offline` to turn off syncing. -wandb: Resuming run qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at https://wandb.ai/wutang/wayfarer-ab-test -wandb: 🚀 View run at https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: Initializing weave. -wandb: uploading summary; updating run metadata -wandb: uploading summary; uploading code/scripts/training/train_serverless_30b_long.py; uploading diff.patch; uploading diff_ca2192b7952143d73db0bde5c2135412ea16d796.patch; uploading wandb-metadata.json (+ 3 more) -wandb: uploading summary -wandb: uploading data -wandb: uploading summary, console lines 6-6 -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-1832 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_183309-qwen3-30b-serverless-rl-long-20260723-1832/logs -File has 19208 rows -Training 100 trajectories -Batches: 50, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 100 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-1832-sft-data-f98e2f90c22a:v0 -Starting SFT training job... - train sft: 0%| | 0/50 [00:00... - training_metrics.append(metrics) - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/serverless/backend.py", line 365, in _train_model - raise RuntimeError(f"Training job failed: {error_message}") -RuntimeError: Training job failed: RuntimeError: Trainable Choice is missing vLLM prompt_token_ids/token_ids. Use a vLLM endpoint with return_token_ids enabled. -Auto-restarting in 60s... -=== Training attempt #2 === -Loading model... -W&B step: 1, checkpoint: 0, resuming from: 1 -Starting SFT Warmup (50 steps)... -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: uploading data -wandb: uploading summary, console lines 116-129 -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 50 -wandb: data/cum/num_trajectories 100 -wandb: data/cum/trainer_tokens 766.44 -wandb: data/step_num_gradient_steps 50 -wandb: data/step_num_trajectories 100 -wandb: data/step_trainer_tokens 766.44 -wandb: loss/grad_norm 6.8367 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 1.43622 -wandb: throughput/avg_trainer_tok_per_s 2.28224 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-1832 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_183842-qwen3-30b-serverless-rl-long-20260723-1832/logs -File has 19208 rows -Training 98 trajectories -Batches: 49, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 98 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-1832-sft-data-fcef31ac60c6:v0 -Starting SFT training job... - train sft: 0%| | 0/49 [00:00... - training_metrics.append(metrics) - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/serverless/backend.py", line 365, in _train_model - raise RuntimeError(f"Training job failed: {error_message}") -RuntimeError: Training job failed: RuntimeError: Trainable Choice is missing vLLM prompt_token_ids/token_ids. Use a vLLM endpoint with return_token_ids enabled. -Auto-restarting in 120s... -=== Training attempt #3 === -Loading model... -W&B step: 2, checkpoint: 0, resuming from: 2 -Starting SFT Warmup (50 steps)... -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: updating run metadata -wandb: uploading wandb-summary.json -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 99 -wandb: data/cum/num_trajectories 198 -wandb: data/cum/trainer_tokens 1539.62367 -wandb: data/step_num_gradient_steps 49 -wandb: data/step_num_trajectories 98 -wandb: data/step_trainer_tokens 773.18367 -wandb: loss/grad_norm 3.62664 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 1.07464 -wandb: throughput/avg_trainer_tok_per_s 2.44885 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-1832 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_190101-qwen3-30b-serverless-rl-long-20260723-1832/logs -File has 19208 rows -Training 96 trajectories -Batches: 48, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 96 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-1832-sft-data-e83609cdc479:v0 -Starting SFT training job... - train sft: 0%| | 0/48 [00:00... - training_metrics.append(metrics) - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/serverless/backend.py", line 365, in _train_model - raise RuntimeError(f"Training job failed: {error_message}") -RuntimeError: Training job failed: RuntimeError: Trainable Choice is missing vLLM prompt_token_ids/token_ids. Use a vLLM endpoint with return_token_ids enabled. -Auto-restarting in 240s... -=== Training attempt #4 === -Loading model... -W&B step: 3, checkpoint: 0, resuming from: 3 -Starting SFT Warmup (50 steps)... -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: updating run metadata -wandb: uploading output.log; uploading wandb-summary.json; uploading config.yaml -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 147 -wandb: data/cum/num_trajectories 294 -wandb: data/cum/trainer_tokens 2299.16534 -wandb: data/step_num_gradient_steps 48 -wandb: data/step_num_trajectories 96 -wandb: data/step_trainer_tokens 759.54167 -wandb: loss/grad_norm 2.4316 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 0.9396 -wandb: throughput/avg_trainer_tok_per_s 2.93921 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-1832 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_192157-qwen3-30b-serverless-rl-long-20260723-1832/logs -File has 19208 rows -Training 94 trajectories -Batches: 47, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 94 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-1832-sft-data-18b1071d069a:v0 -Starting SFT training job... - train sft: 0%| | 0/47 [00:00... - training_metrics.append(metrics) - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/serverless/backend.py", line 365, in _train_model - raise RuntimeError(f"Training job failed: {error_message}") -RuntimeError: Training job failed: RuntimeError: Trainable Choice is missing vLLM prompt_token_ids/token_ids. Use a vLLM endpoint with return_token_ids enabled. -Auto-restarting in 300s... -=== Training attempt #5 === -Loading model... -W&B step: 4, checkpoint: 0, resuming from: 4 -Starting SFT Warmup (50 steps)... -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: uploading summary -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 194 -wandb: data/cum/num_trajectories 388 -wandb: data/cum/trainer_tokens 3067.25045 -wandb: data/step_num_gradient_steps 47 -wandb: data/step_num_trajectories 94 -wandb: data/step_trainer_tokens 768.08511 -wandb: loss/grad_norm 2.21089 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 0.87397 -wandb: throughput/avg_trainer_tok_per_s 3.3125 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-1832 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-1832 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_194442-qwen3-30b-serverless-rl-long-20260723-1832/logs -File has 19208 rows -Training 92 trajectories -Batches: 46, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 92 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-1832-sft-data-2a005b08d1ca:v0 -Starting SFT training job... - train sft: 0%| | 0/46 [00:00... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 50, in gather_trajectory_groups - result_groups = await future - ^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 41, in group_forward - group = await wrap_group_awaitable(g) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 144, in wrap_group_awaitable - return await awaitable - ^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/trajectories.py", line 282, in _ - trajectory = await future - ^^^^^^^^^^^^ - File "/data/vivi/.local/share/uv/python/cpython-3.13.14-linux-x86_64-gnu/lib/python3.13/asyncio/tasks.py", line 634, in _wait_for_one - return f.result() if resolve else f - ~~~~~~~~^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Auto-restarting in 60s... - RL step 1: 0%| | 0/40 [00:00 exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: updating run metadata -wandb: uploading summary, console lines 154-1708 -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 50 -wandb: data/cum/num_trajectories 100 -wandb: data/cum/trainer_tokens 766.44 -wandb: data/step_num_gradient_steps 50 -wandb: data/step_num_trajectories 100 -wandb: data/step_trainer_tokens 766.44 -wandb: loss/grad_norm 6.84095 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 1.43631 -wandb: throughput/avg_trainer_tok_per_s 5.07079 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-2013 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-2013 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_201637-qwen3-30b-serverless-rl-long-20260723-2013/logs -File has 19208 rows -Training 98 trajectories -Batches: 49, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 98 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-2013-sft-data-95cd621d18b5:v0 -Starting SFT training job... - train sft: 0%| | 0/49 [00:00... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 50, in gather_trajectory_groups - result_groups = await future - ^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 41, in group_forward - group = await wrap_group_awaitable(g) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/gather.py", line 144, in wrap_group_awaitable - return await awaitable - ^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/art/trajectories.py", line 282, in _ - trajectory = await future - ^^^^^^^^^^^^ - File "/data/vivi/.local/share/uv/python/cpython-3.13.14-linux-x86_64-gnu/lib/python3.13/asyncio/tasks.py", line 634, in _wait_for_one - return f.result() if resolve else f - ~~~~~~~~^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Auto-restarting in 120s... -weave: 🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8ba3-7f2a-aab2-105d1255a51b -🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8ba3-7f2a-aab2-105d1255a51b -weave: 🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8bc2-79f1-b09c-d563c8c4dfef -🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8bc2-79f1-b09c-d563c8c4dfef -weave: 🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8bb3-7392-930c-f3e09680b086 -🍩 https://wandb.ai/wutang/wayfarer-ab-test/r/call/019f90a2-8bb3-7392-930c-f3e09680b086 - RL step 2: 0%| | 0/40 [00:00 exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -Task exception was never retrieved -future: exception=BadRequestError("Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}}")> -Traceback (most recent call last): - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/data/vivi/pixelated/scripts/training/serverless_utils.py", line 70, in rollout - completion = await azure_client.chat.completions.create( - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ...<6 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 1298, in wrapper - res, _ = await _call_async_func( - ^^^^^^^^^^^^^^^^^^^^^^^ - cast(Op[P, R], wrapper), *args, __should_raise=True, **kwargs - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/trace/op.py", line 692, in _call_async_func - res = await func(*args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/weave/integrations/openai/openai_sdk.py", line 501, in _wrapper - return await fn(self, *args, **kwargs) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/resources/chat/completions/completions.py", line 2877, in create - return await self._post( - ^^^^^^^^^^^^^^^^^ - ...<55 lines>... - ) - ^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1931, in post - return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls) - ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ - File "/home/vivi/pixelated/.venv/lib/python3.13/site-packages/openai/_base_client.py", line 1716, in request - raise self._make_status_error_from_response(err.response) from None -openai.BadRequestError: Error code: 400 - {'error': {'code': 'unrecognized_request_argument', 'message': 'Unrecognized request argument supplied: return_token_ids', 'details': 'Unrecognized request argument supplied: return_token_ids'}} -wandb: WARNING [wandb.login()] Changing session credentials to explicit value for https://api.wandb.ai. -wandb: wandb.init() called while a run is active and reinit is set to 'default', so returning the previous run. -wandb: Initializing weave. -wandb: updating run metadata -wandb: uploading config.yaml; uploading output.log; uploading wandb-summary.json -wandb: uploading summary, console lines 151-1705 -wandb: -wandb: Run history: -wandb: data/cum/num_gradient_steps ▁ -wandb: data/cum/num_trajectories ▁ -wandb: data/cum/trainer_tokens ▁ -wandb: data/step_num_gradient_steps ▁ -wandb: data/step_num_trajectories ▁ -wandb: data/step_trainer_tokens ▁ -wandb: loss/grad_norm ▁ -wandb: loss/learning_rate ▁ -wandb: loss/train ▁ -wandb: throughput/avg_trainer_tok_per_s ▁ -wandb: +7 ... -wandb: -wandb: Run summary: -wandb: data/cum/num_gradient_steps 99 -wandb: data/cum/num_trajectories 198 -wandb: data/cum/trainer_tokens 1539.62367 -wandb: data/step_num_gradient_steps 49 -wandb: data/step_num_trajectories 98 -wandb: data/step_trainer_tokens 773.18367 -wandb: loss/grad_norm 3.61805 -wandb: loss/learning_rate 1e-05 -wandb: loss/train 1.07502 -wandb: throughput/avg_trainer_tok_per_s 5.48587 -wandb: +7 ... -wandb: -wandb: 🚀 View run qwen3-30b-serverless-rl-long-20260723-2013 at: https://wandb.ai/wutang/wayfarer-ab-test/runs/qwen3-30b-serverless-rl-long-20260723-2013 -wandb: ⭐️ View project at: https://wandb.ai/wutang/wayfarer-ab-test -wandb: Synced 8 W&B file(s), 0 media file(s), 2 artifact file(s) and 0 other file(s) -wandb: Find logs at: ./wandb/run-20260723_201954-qwen3-30b-serverless-rl-long-20260723-2013/logs -File has 19208 rows -Training 96 trajectories -Batches: 48, batch_size: 2 -Schedule: linear, peak_lr: 0.0002 -Serializing trajectories to file (streaming)... -Serialized 96 trajectories -Uploading training data to W&B artifacts... -Training data uploaded. Artifact URL: wandb-artifact:///wutang/wayfarer-ab-test/qwen3-30b-serverless-rl-long-20260723-2013-sft-data-02674d066d52:v0 -Starting SFT training job... - train sft: 0%| | 0/48 [00:00 - - - - - AI Security Test Report - - - - -

AI Security Test Report

- -
-

Executive Summary

-

- Security test report generated on 3/17/2025, 7:50:42 PM for environment: - test -

- -
-
-

Total Tests

-
3
-
-
-

Critical Issues

-
0
-
-
-

High Issues

-
3
-
-
-

Total Duration

-
0.00s
-
-
- -
- -
-
- -
-

Endpoint Security

-

Tests AI endpoints for basic security issues

- - -
-

Endpoint Security

-

- Test suite Endpoint Security failed with code 1: TypeError: Unknown - file extension ".ts" for - /Users/vivi/astro/tests/security/ai-endpoint-scanner.ts at - Object.getFileProtocolModuleFormat [as file:] - (node:internal/modules/esm/get_format:219:9) at defaultGetFormat - (node:internal/modules/esm/get_format:245:36) at defaultLoad - (node:internal/modules/esm/load:120:22) at async - ModuleLoader.loadAndTranslate - (node:internal/modules/esm/loader:483:32) at async ModuleJob.link - (node:internal/modules/esm/module_job:115:19) { code: - 'ERR_UNKNOWN_FILE_EXTENSION' } -

- - -
-
- -
-

Authentication Bypass

-

Tests for authentication and authorization bypass vulnerabilities

- - -
-

Authentication Bypass

-

- Test suite Authentication Bypass failed with code 1: TypeError: - Unknown file extension ".ts" for - /Users/vivi/astro/tests/security/ai-auth-bypass-tester.ts at - Object.getFileProtocolModuleFormat [as file:] - (node:internal/modules/esm/get_format:219:9) at defaultGetFormat - (node:internal/modules/esm/get_format:245:36) at defaultLoad - (node:internal/modules/esm/load:120:22) at async - ModuleLoader.loadAndTranslate - (node:internal/modules/esm/loader:483:32) at async ModuleJob.link - (node:internal/modules/esm/module_job:115:19) { code: - 'ERR_UNKNOWN_FILE_EXTENSION' } -

- - -
-
- -
-

Web Vulnerabilities

-

Tests for common web vulnerabilities

- - -
-

Web Vulnerabilities

-

- Test suite Web Vulnerabilities failed with code 1: TypeError: Unknown - file extension ".ts" for - /Users/vivi/astro/tests/security/ai-web-vulnerability-scanner.ts at - Object.getFileProtocolModuleFormat [as file:] - (node:internal/modules/esm/get_format:219:9) at defaultGetFormat - (node:internal/modules/esm/get_format:245:36) at defaultLoad - (node:internal/modules/esm/load:120:22) at async - ModuleLoader.loadAndTranslate - (node:internal/modules/esm/loader:483:32) at async ModuleJob.link - (node:internal/modules/esm/module_job:115:19) { code: - 'ERR_UNKNOWN_FILE_EXTENSION' } -

- - -
-
- - - - diff --git a/tests/security/security-reports/ai-security-report-2025-03-18.html b/tests/security/security-reports/ai-security-report-2025-03-18.html deleted file mode 100644 index f7a954ffb9..0000000000 --- a/tests/security/security-reports/ai-security-report-2025-03-18.html +++ /dev/null @@ -1,229 +0,0 @@ - - - - - - AI Security Test Report - - - - -

AI Security Test Report

- -
-

Executive Summary

-

- Security test report generated on 3/17/2025, 11:09:20 PM for - environment: test -

- -
-
-

Total Tests

-
3
-
-
-

Critical Issues

-
0
-
-
-

High Issues

-
0
-
-
-

Total Duration

-
33.60s
-
-
- -
- -
-
- -
-

Endpoint Security

-

Tests AI endpoints for basic security issues

- - -

Error parsing test results

-
- -
-

Authentication Bypass

-

Tests for authentication and authorization bypass vulnerabilities

- - -

Error parsing test results

-
- -
-

Web Vulnerabilities

-

Tests for common web vulnerabilities

- - -

Error parsing test results

-
- - - - diff --git a/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-17.json b/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-17.json deleted file mode 100644 index cd01881c4c..0000000000 --- a/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-17.json +++ /dev/null @@ -1,29 +0,0 @@ -[ - { - "testName": "Security Headers Test", - "category": "Security Headers", - "severity": "High", - "vulnerable": true, - "details": "Error testing security headers: ", - "responseTime": 13.295041999999967, - "timestamp": "2025-03-17T23:38:52.672Z" - }, - { - "testName": "CORS Test", - "category": "CORS", - "severity": "High", - "vulnerable": true, - "details": "Error testing CORS: ", - "responseTime": 2.1250420000000076, - "timestamp": "2025-03-17T23:38:53.175Z" - }, - { - "testName": "Information Disclosure Test", - "category": "Information Disclosure", - "severity": "High", - "vulnerable": true, - "details": "Error testing information disclosure: ", - "responseTime": 10.40645799999993, - "timestamp": "2025-03-17T23:38:53.688Z" - } -] diff --git a/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-18.json b/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-18.json deleted file mode 100644 index eaed553a7c..0000000000 --- a/tests/security/security-reports/ai-web-vulnerability-scan-2025-03-18.json +++ /dev/null @@ -1,29 +0,0 @@ -[ - { - "testName": "Security Headers Test", - "category": "Security Headers", - "severity": "High", - "vulnerable": true, - "details": "Error testing security headers: ", - "responseTime": 13.910708, - "timestamp": "2025-03-18T03:09:19.117Z" - }, - { - "testName": "CORS Test", - "category": "CORS", - "severity": "High", - "vulnerable": true, - "details": "Error testing CORS: ", - "responseTime": 2.764499999999998, - "timestamp": "2025-03-18T03:09:19.621Z" - }, - { - "testName": "Information Disclosure Test", - "category": "Information Disclosure", - "severity": "High", - "vulnerable": true, - "details": "Error testing information disclosure: ", - "responseTime": 17.474083000000064, - "timestamp": "2025-03-18T03:09:20.141Z" - } -] diff --git a/tests/temp/test-dashboard-fix.js b/tests/temp/test-dashboard-fix.js deleted file mode 100644 index a0d7fd7563..0000000000 --- a/tests/temp/test-dashboard-fix.js +++ /dev/null @@ -1,29 +0,0 @@ -import { chromium } from 'playwright' - -async function testDashboard() { - const browser = await chromium.launch() - const page = await browser.newPage() - - console.log('Navigating to dashboard...') - await page.goto('http://localhost:4321/dashboard') - - console.log('Current URL:', page.url()) - - // Check if we have the sidebar elements - const sidebar = await page.locator('.sidebar, .dashboard-sidebar').count() - console.log('Sidebar elements found:', sidebar) - - if (sidebar > 0) { - console.log('✅ Test PASSED: Sidebar elements are present') - } else { - console.log('❌ Test FAILED: No sidebar elements found') - } - - await browser.close() - process.exit(sidebar > 0 ? 0 : 1) -} - -testDashboard().catch((error) => { - console.error('Test error:', error) - process.exit(1) -}) diff --git a/tests/test-results/e2e-report/index.html b/tests/test-results/e2e-report/index.html deleted file mode 100644 index 11a2af8783..0000000000 --- a/tests/test-results/e2e-report/index.html +++ /dev/null @@ -1,19421 +0,0 @@ - - - - - - - Playwright Test Report - - - - -
- - - diff --git a/tests/test-results/e2e-results.json b/tests/test-results/e2e-results.json deleted file mode 100644 index a1c8f95d8e..0000000000 --- a/tests/test-results/e2e-results.json +++ /dev/null @@ -1,88 +0,0 @@ -{ - "config": { - "configFile": "~/astro/tests/e2e-config.ts", - "rootDir": "~/astro/tests/tests/e2e", - "forbidOnly": false, - "fullyParallel": false, - "globalSetup": null, - "globalTeardown": null, - "globalTimeout": 0, - "grep": {}, - "grepInvert": null, - "maxFailures": 0, - "metadata": {}, - "preserveOutput": "always", - "reporter": [ - [ - "html", - { - "open": "on-failure", - "outputFolder": "test-results/e2e-report" - } - ], - [ - "json", - { - "outputFile": "test-results/e2e-results.json" - } - ] - ], - "reportSlowTests": { - "max": 5, - "threshold": 300000 - }, - "quiet": false, - "projects": [ - { - "outputDir": "~/astro/test-results", - "repeatEach": 1, - "retries": 0, - "metadata": {}, - "id": "chromium", - "name": "chromium", - "testDir": "~/astro/tests/tests/e2e", - "testIgnore": [], - "testMatch": ["**/*.@(spec|test).?(c|m)[jt]s?(x)"], - "timeout": 60000 - }, - { - "outputDir": "~/astro/test-results", - "repeatEach": 1, - "retries": 0, - "metadata": {}, - "id": "Mobile Chrome", - "name": "Mobile Chrome", - "testDir": "~/astro/tests/tests/e2e", - "testIgnore": [], - "testMatch": ["**/*.@(spec|test).?(c|m)[jt]s?(x)"], - "timeout": 60000 - } - ], - "shard": null, - "updateSnapshots": "missing", - "updateSourceMethod": "patch", - "version": "1.51.1", - "workers": 2, - "webServer": { - "command": "pnpm run dev", - "port": 3000, - "reuseExistingServer": true, - "timeout": 60000 - } - }, - "suites": [], - "errors": [ - { - "message": "Error: No tests found", - "stack": "Error: No tests found" - } - ], - "stats": { - "startTime": "2025-04-12T11:03:00.182Z", - "duration": 11018.396, - "expected": 0, - "skipped": 0, - "unexpected": 0, - "flaky": 0 - } -} diff --git a/tests/unit/ai/test_preprocessing_pipeline.py.bak b/tests/unit/ai/test_preprocessing_pipeline.py.bak deleted file mode 100644 index 7c605498de..0000000000 --- a/tests/unit/ai/test_preprocessing_pipeline.py.bak +++ /dev/null @@ -1,225 +0,0 @@ -""" -Tests for the multimodal preprocessing pipeline. -""" - -import asyncio -import base64 -import json -import tempfile -from pathlib import Path -from unittest.mock import AsyncMock, Mock, patch - -import numpy as np -import pytest -from PIL import Image - -from ..preprocessing_pipeline import ( - AudioPreprocessingPipeline, - VisionPreprocessingPipeline, - VideoPreprocessingPipeline, - MultimodalPreprocessingPipeline, - validate_audio_format, - validate_image_format, - validate_video_format -) - - -class TestAudioPreprocessingPipeline: - """Test audio preprocessing pipeline""" - - @pytest.fixture - def audio_pipeline(self): - """Create audio preprocessing pipeline instance""" - return AudioPreprocessingPipeline() - - @pytest.fixture - def sample_audio_bytes(self): - """Create sample audio bytes (mock data)""" - # Create mock audio data - audio_data = np.random.randn(16000).astype(np.float32) # 1 second of random audio - return audio_data.tobytes() - - @pytest.mark.asyncio - async def test_initialize_pipeline(self, audio_pipeline): - """Test pipeline initialization""" - assert audio_pipeline is not None - assert audio_pipeline.is_loaded == False - - @pytest.mark.asyncio - async def test_load_processors(self, audio_pipeline): - """Test loading processors""" - with patch('transformers.WhisperProcessor.from_pretrained') as mock_whisper, \ - patch('transformers.Wav2Vec2Processor.from_pretrained') as mock_wav2vec: - - mock_whisper.return_value = Mock() - mock_wav2vec.return_value = Mock() - - result = await audio_pipeline.load_processors() - assert result == True - assert audio_pipeline.is_loaded == True - - @pytest.mark.asyncio - async def test_preprocess_audio_bytes(self, audio_pipeline, sample_audio_bytes): - """Test preprocessing audio from bytes""" - with patch.object(audio_pipeline, 'load_processors', AsyncMock()), \ - patch('librosa.load') as mock_librosa, \ - patch.object(audio_pipeline, '_extract_audio_features', AsyncMock(return_value={})), \ - patch.object(audio_pipeline, '_prepare_model_inputs', AsyncMock(return_value={})): - - # Mock librosa.load to return sample data - mock_librosa.return_value = (np.array([0.1, 0.2, 0.3]), 16000) - - result = await audio_pipeline.preprocess_audio(sample_audio_bytes) - - assert "audio_array" in result - assert "sample_rate" in result - assert "duration" in result - assert result["sample_rate"] == 16000 - - @pytest.mark.asyncio - async def test_normalize_audio(self, audio_pipeline): - """Test audio normalization""" - test_audio = np.array([2.0, -3.0, 1.0, -1.0]) - normalized = audio_pipeline._normalize_audio(test_audio) - - # Check that values are in [-1, 1] range - assert np.all(normalized <= 1.0) - assert np.all(normalized >= -1.0) - - # Check that max absolute value is 1.0 - assert np.max(np.abs(normalized)) == 1.0 - - -class TestVisionPreprocessingPipeline: - """Test vision preprocessing pipeline""" - - @pytest.fixture - def vision_pipeline(self): - """Create vision preprocessing pipeline instance""" - return VisionPreprocessingPipeline() - - @pytest.fixture - def sample_image_bytes(self): - """Create sample image bytes""" - # Create a simple test image - image = Image.new('RGB', (100, 100), color='red') - with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp: - image.save(tmp, format='PNG') - tmp_path = tmp.name - - with open(tmp_path, 'rb') as f: - image_bytes = f.read() - - Path(tmp_path).unlink() - return image_bytes - - @pytest.mark.asyncio - async def test_initialize_pipeline(self, vision_pipeline): - """Test pipeline initialization""" - assert vision_pipeline is not None - assert vision_pipeline.is_loaded == False - - @pytest.mark.asyncio - async def test_load_processor(self, vision_pipeline): - """Test loading processor""" - with patch('transformers.ViTImageProcessor.from_pretrained') as mock_vit: - mock_vit.return_value = Mock() - - result = await vision_pipeline.load_processor() - assert result == True - assert vision_pipeline.is_loaded == True - - @pytest.mark.asyncio - async def test_preprocess_image_bytes(self, vision_pipeline, sample_image_bytes): - """Test preprocessing image from bytes""" - with patch.object(vision_pipeline, 'load_processor', AsyncMock()), \ - patch.object(vision_pipeline, '_prepare_model_inputs', AsyncMock(return_value={})): - - result = await vision_pipeline.preprocess_image(sample_image_bytes) - - assert "image" in result - assert "width" in result - assert "height" in result - assert result["width"] == 100 - assert result["height"] == 100 - - -class TestVideoPreprocessingPipeline: - """Test video preprocessing pipeline""" - - @pytest.fixture - def video_pipeline(self): - """Create video preprocessing pipeline instance""" - return VideoPreprocessingPipeline() - - @pytest.mark.asyncio - async def test_initialize_pipeline(self, video_pipeline): - """Test pipeline initialization""" - assert video_pipeline is not None - assert video_pipeline.is_loaded == False - - -class TestMultimodalPreprocessingPipeline: - """Test multimodal preprocessing pipeline""" - - @pytest.fixture - def multimodal_pipeline(self): - """Create multimodal preprocessing pipeline instance""" - return MultimodalPreprocessingPipeline() - - @pytest.mark.asyncio - async def test_initialize_pipeline(self, multimodal_pipeline): - """Test pipeline initialization""" - assert multimodal_pipeline is not None - assert multimodal_pipeline.audio_pipeline is not None - assert multimodal_pipeline.vision_pipeline is not None - assert multimodal_pipeline.video_pipeline is not None - - -class TestValidationFunctions: - """Test data validation functions""" - - def test_validate_audio_format(self): - """Test audio format validation""" - # Test valid formats - assert validate_audio_format("test.mp3") == True - assert validate_audio_format("test.wav") == True - assert validate_audio_format("test.flac") == True - - # Test invalid format - assert validate_audio_format("test.txt") == False - - def test_validate_image_format(self): - """Test image format validation""" - # Test valid formats - assert validate_image_format("test.jpg") == True - assert validate_image_format("test.png") == True - assert validate_image_format("test.gif") == True - - # Test invalid format - assert validate_image_format("test.mp3") == False - - def test_validate_video_format(self): - """Test video format validation""" - # Test valid formats - assert validate_video_format("test.mp4") == True - assert validate_video_format("test.avi") == True - assert validate_video_format("test.mov") == True - - # Test invalid format - assert validate_video_format("test.jpg") == False - - -class TestIntegration: - """Integration tests for preprocessing pipeline""" - - @pytest.mark.asyncio - async def test_full_preprocessing_pipeline(self): - """Test full preprocessing pipeline integration""" - # This would test the complete flow with mocked dependencies - pipeline = MultimodalPreprocessingPipeline() - assert pipeline is not None - - -if __name__ == "__main__": - pytest.main([__file__]) \ No newline at end of file