chore: format

This commit is contained in:
Timothy Jaeryang Baek
2025-03-01 07:28:00 -08:00
parent d7088efe73
commit efe8c4ca69
53 changed files with 397 additions and 7 deletions

View File

@@ -619,7 +619,9 @@ async def update_rag_config(
request.app.state.config.EXA_API_KEY = form_data.web.search.exa_api_key
request.app.state.config.PERPLEXITY_API_KEY = form_data.web.search.perplexity_api_key
request.app.state.config.PERPLEXITY_API_KEY = (
form_data.web.search.perplexity_api_key
)
request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT = (
form_data.web.search.result_count

View File

@@ -452,13 +452,14 @@ def parse_ollama_modelfile(model_text):
return data
def convert_logit_bias_input_to_json(user_input):
logit_bias_pairs = user_input.split(',')
logit_bias_pairs = user_input.split(",")
logit_bias_json = {}
for pair in logit_bias_pairs:
token, bias = pair.split(':')
token, bias = pair.split(":")
token = str(token.strip())
bias = int(bias.strip())
bias = 100 if bias > 100 else -100 if bias < -100 else bias
logit_bias_json[token] = bias
return json.dumps(logit_bias_json)
return json.dumps(logit_bias_json)