Trying some stuff, including a new model
This commit is contained in:
parent
20688ea3b4
commit
6ea0e48cc3
|
@ -20,7 +20,8 @@ from . import tools
|
||||||
cli_history = prompt_toolkit.history.FileHistory('output/cli_history.txt')
|
cli_history = prompt_toolkit.history.FileHistory('output/cli_history.txt')
|
||||||
|
|
||||||
#MODEL = "gemma3:27b"
|
#MODEL = "gemma3:27b"
|
||||||
MODEL = "qwen3:latest"
|
#MODEL = "qwen3:latest"
|
||||||
|
MODEL = 'hf.co/unsloth/Qwen3-30B-A3B-GGUF:Q4_K_M'
|
||||||
|
|
||||||
def create_model():
|
def create_model():
|
||||||
available_tools = tools.get_tools()
|
available_tools = tools.get_tools()
|
||||||
|
|
|
@ -7,6 +7,7 @@ from typing import get_type_hints, List, Dict, Any
|
||||||
from collections.abc import Iterator
|
from collections.abc import Iterator
|
||||||
import logging
|
import logging
|
||||||
import secret_loader
|
import secret_loader
|
||||||
|
from decimal import Decimal
|
||||||
|
|
||||||
try:
|
try:
|
||||||
import pycountry.db
|
import pycountry.db
|
||||||
|
@ -25,14 +26,18 @@ def search(query: str):
|
||||||
|
|
||||||
def dataclasses_to_json(data):
|
def dataclasses_to_json(data):
|
||||||
if pycountry and isinstance(data, pycountry.db.Country):
|
if pycountry and isinstance(data, pycountry.db.Country):
|
||||||
return data.alpha_3
|
return data.alpha_2
|
||||||
if isinstance(data, list | tuple):
|
if isinstance(data, list | tuple):
|
||||||
return [dataclasses_to_json(d) for d in data]
|
return [dataclasses_to_json(d) for d in data]
|
||||||
|
if isinstance(data, dict):
|
||||||
|
return {k:dataclasses_to_json(v) for k,v in data.items() if v}
|
||||||
|
if isinstance(data, Decimal):
|
||||||
|
return float(data)
|
||||||
if dataclasses.is_dataclass(data):
|
if dataclasses.is_dataclass(data):
|
||||||
result = {}
|
result = {}
|
||||||
for field in dataclasses.fields(data):
|
for field in dataclasses.fields(data):
|
||||||
result[field.name] = dataclasses_to_json(getattr(data,field.name))
|
result[field.name] = getattr(data,field.name)
|
||||||
return result
|
return dataclasses_to_json(result)
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user