Aurora-Code-1

pip install "transformers>=4.51.0" accelerate

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "Perciqa/Aurora-Code-1"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype="auto",
    device_map="auto",
)

system_prompt = (
    "You are Aurora, an AI code assistant built by Perciqa. "
    "You help developers write, review, and understand code. "
    "You provide clear, correct, and complete solutions. "
    "When you're unsure, you say so."
)

messages = [
    {"role": "system", "content": system_prompt},
    {"role": "user", "content": "Write a Python function to merge two sorted lists."},
]

text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer([text], return_tensors="pt").to(model.device)

outputs = model.generate(
    **inputs,
    max_new_tokens=2048,
    temperature=0.7,
    do_sample=True,
)
response = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)
print(response)

pip install vllm
vllm serve Perciqa/Aurora-Code-1 --max-model-len 32768

from openai import OpenAI

client = OpenAI(base_url="http://localhost:8000/v1", api_key="token-abc123")

response = client.chat.completions.create(
    model="Perciqa/Aurora-Code-1",
    messages=[
        {"role": "system", "content": "You are Aurora, an AI code assistant built by Perciqa."},
        {"role": "user", "content": "Refactor this function to be more Pythonic."},
    ],
    max_tokens=2048,
)
print(response.choices[0].message.content)

ollama run hf.co/Perciqa/Aurora-Code-1

import argus
from openai import OpenAI

argus.init(server_url="http://localhost:8000", agent_name="aurora-coder")
client = OpenAI(base_url="http://localhost:8000/v1", api_key="token-abc123")

@argus.trace(kind="agent")
def aurora_code(query: str) -> str:
    response = client.chat.completions.create(
        model="Perciqa/Aurora-Code-1",
        messages=[
            {"role": "system", "content": "You are Aurora, an AI code assistant built by Perciqa."},
            {"role": "user", "content": query},
        ],
    )
    return response.choices[0].message.content

result = aurora_code("Write a TypeScript function that validates an email address.")
print(result)

You are Aurora, an AI code assistant built by Perciqa. You help developers write, review, and understand code. You provide clear, correct, and complete solutions. When you're unsure, you say so.