File size: 3,547 Bytes
2695555 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | from _util import carregar_tars
tars = carregar_tars()
catalogo_fake = {
"qwen3:32b": {"tag": "qwen3:32b", "familia": "qwen3", "categoria": "tecnico", "parametros_b": 32, "capacidades": {"tools"}, "capacidades_conhecidas": True},
"qwen3.8:27b": {"tag": "qwen3.8:27b", "familia": "qwen3", "categoria": "visao", "parametros_b": 27, "capacidades": {"tools", "vision"}, "capacidades_conhecidas": True},
"qwen3:0.6b": {"tag": "qwen3:0.6b", "familia": "qwen3", "categoria": "simples", "parametros_b": 0.6, "capacidades": {"tools"}, "capacidades_conhecidas": True},
"qwen3:8b": {"tag": "qwen3:8b", "familia": "qwen3", "categoria": "geral", "parametros_b": 8, "capacidades": {"tools"}, "capacidades_conhecidas": True},
"llama3.1:latest": {"tag": "llama3.1:latest", "familia": "llama", "categoria": "geral", "parametros_b": 8, "capacidades": {"tools"}, "capacidades_conhecidas": True},
"gemma3:270m": {"tag": "gemma3:270m", "familia": "gemma3", "categoria": "simples", "parametros_b": 0.27, "capacidades": set(), "capacidades_conhecidas": True},
}
tars.catalogar_modelos = lambda forcar=False: catalogo_fake
tars.listar_modelos_instalados = lambda forcar=False: set(catalogo_fake.keys())
tars._carregar_em_segundo_plano = lambda modelo: None
def sem_chamada_ia(*a, **k):
raise AssertionError("nao deveria precisar da IA classificadora pra esse caso (deveria casar por texto)")
tars._SESSION.post = sem_chamada_ia
# Caso do bug relatado: "/modelo gemma 270m"
assert tars.interpretar_comando_modelo("/modelo gemma 270m") is True
assert tars.modelo_manual == "gemma3:270m", tars.modelo_manual
assert tars.modo_modelo == "MANUAL"
print("OK: '/modelo gemma 270m' agora resolve para gemma3:270m")
# Reset
tars.modo_modelo = "AUTO"; tars.modelo_manual = None
# Variações de digitação — sempre via "/modelo", que é agora a ÚNICA
# forma de trocar por texto livre/aproximado. Trocar a partir de um
# verbo solto numa frase qualquer ("usar o gemma", "troca pra qwen")
# foi removido de propósito (ver test_sem_linguagem_natural.py).
for entrada, esperado in [
("/modelo gemma", "gemma3:270m"),
("/modelo qwen 8b", "qwen3:8b"),
("/modelo llama", "llama3.1:latest"),
("/modelo qwen 32b", "qwen3:32b"),
("/modelo usar o gemma 270m", "gemma3:270m"),
("/modelo troca pra qwen 8b por favor", "qwen3:8b"),
]:
tars.modo_modelo = "AUTO"; tars.modelo_manual = None
ok = tars.interpretar_comando_modelo(entrada)
status = "OK" if (ok and tars.modelo_manual == esperado) else "FALHOU"
print(f"{status:6} {entrada!r:35} -> {tars.modelo_manual} (esperado {esperado})")
assert ok and tars.modelo_manual == esperado
# Mensagem inteira igual a uma tag/família/alias continua funcionando
# SEM precisar de "/modelo" (é a frase toda, sem ambiguidade nenhuma).
# "llama" é a família reportada (e única) desse modelo no catálogo.
tars.modo_modelo = "AUTO"; tars.modelo_manual = None
assert tars.interpretar_comando_modelo("llama") is True
assert tars.modelo_manual == "llama3.1:latest"
print("OK: mensagem inteira == alias (família) continua trocando sem /modelo")
# Caso: pedido que não bate em nada -> avisa, nao muda o modo, nao quebra
tars.modo_modelo = "AUTO"; tars.modelo_manual = None
ok = tars.interpretar_comando_modelo("/modelo bagulho-que-nao-existe-999")
assert ok is True # comando reconhecido (é /modelo), so nao achou correspondencia
assert tars.modo_modelo == "AUTO" # nao mudou nada
print("OK: modelo inexistente avisa sem travar nem mudar o estado")
print("\nTUDO OK.")
|