26F25d

import argparse
from ollama import Client

def main():
    parser = argparse.ArgumentParser(description="Ollama 원격 호출 도구")
    # 필수 파라미터: 질문
    parser.add_argument("prompt", type=str, help="LLM에게 던질 질문")
    # 선택 파라미터: 모델 (기본값 설정 가능)
    parser.add_argument("--model", default="llama3.3", help="사용할 모델명")
    
    args = parser.parse_args()

    client = Client(host='http://<서버_IP>:11434')
    response = client.chat(model=args.model, messages=[
        {'role': 'user', 'content': args.prompt},
    ])
    
    print(f"\n[모델: {args.model}]의 답변:")
    print(response['message']['content'])

if __name__ == "__main__":
    main()