claude-proxy / patch_vision.py
dheraingoud's picture
Deploy latest upstream with key rotation, vision adapter, and HF configs
61bb677
Raw
History Blame Contribute Delete
1.23 kB
with open('src/free_claude_code/providers/nvidia_nim/vision_adapter.py', 'r', encoding='utf-8') as f:
content = f.read()
func = '''
def inject_vision_into_request(request_data: Any) -> None:
\"\"\"Inject vision support by promoting tool result images into standard images.\"\"\"
if hasattr(request_data, 'messages'):
request_data.messages = promote_tool_result_images(request_data.messages)
elif isinstance(request_data, dict) and 'messages' in request_data:
request_data['messages'] = promote_tool_result_images(request_data['messages'])
'''
if 'def inject_vision_into_request' not in content:
content += f'\n\n{func}\n'
with open('src/free_claude_code/providers/nvidia_nim/vision_adapter.py', 'w', encoding='utf-8') as f:
f.write(content)
with open('src/free_claude_code/providers/nvidia_nim/request_options.py', 'r', encoding='utf-8') as f:
opt = f.read()
opt = opt.replace('from .vision_adapter import inject_vision_into_request\nfrom .vision_adapter import inject_vision_into_request', 'from .vision_adapter import inject_vision_into_request')
with open('src/free_claude_code/providers/nvidia_nim/request_options.py', 'w', encoding='utf-8') as f:
f.write(opt)