model-metadata/code_python_files

1# /// script2# requires-python = ">=3.12"3# dependencies = [4#     "torch",5#     "torchvision",6#     "transformers",7#     "accelerate",8#     "peft",9#     "slack-sdk",10# ]11# ///12 13try:14    import os15    from openai import OpenAI16    17    client = OpenAI(18        base_url="https://router.huggingface.co/v1",19        api_key=os.environ["HF_TOKEN"],20    )21    22    completion = client.chat.completions.create(23        model="katanemo/Arch-Router-1.5B",24        messages=[25            {26                "role": "user",27                "content": "What is the capital of France?"28            }29        ],30    )31    32    print(completion.choices[0].message)33    with open('katanemo_Arch-Router-1.5B_4.txt', 'w', encoding='utf-8') as f:34        f.write('Everything was good in katanemo_Arch-Router-1.5B_4.txt')35except Exception as e:36    import os37    from slack_sdk import WebClient38    client = WebClient(token=os.environ['SLACK_TOKEN'])39    client.chat_postMessage(40        channel='#exp-slack-alerts',41        text='Problem in <https://huggingface.co/datasets/model-metadata/code_execution_files/blob/main/katanemo_Arch-Router-1.5B_4.txt|katanemo_Arch-Router-1.5B_4.txt>',42    )43 44    with open('katanemo_Arch-Router-1.5B_4.txt', 'a', encoding='utf-8') as f:45        import traceback46        f.write('''```CODE: 47import os48from openai import OpenAI49 50client = OpenAI(51    base_url="https://router.huggingface.co/v1",52    api_key=os.environ["HF_TOKEN"],53)54 55completion = client.chat.completions.create(56    model="katanemo/Arch-Router-1.5B",57    messages=[58        {59            "role": "user",60            "content": "What is the capital of France?"61        }62    ],63)64 65print(completion.choices[0].message)66```67 68ERROR: 69''')70        traceback.print_exc(file=f)71    72finally:73    from huggingface_hub import upload_file74    upload_file(75        path_or_fileobj='katanemo_Arch-Router-1.5B_4.txt',76        repo_id='model-metadata/code_execution_files',77        path_in_repo='katanemo_Arch-Router-1.5B_4.txt',78        repo_type='dataset',79    )80