fix : server crash with ollama

This commit is contained in:
martin legrand
2025-03-29 18:07:38 +01:00
parent d8ded2d456
commit e0eee90202
4 changed files with 25 additions and 12 deletions
+2 -2
View File
@@ -3,8 +3,8 @@
import argparse
from flask import Flask, jsonify, request
from sources.llamacpp import LlamacppLLM
from sources.ollama import OllamaLLM
from sources.llamacpp_handler import LlamacppLLM
from sources.ollama_handler import OllamaLLM
parser = argparse.ArgumentParser(description='AgenticSeek server script')
parser.add_argument('--provider', type=str, help='LLM backend library to use. set to [ollama] or [llamacpp]', required=True)