25 lines
862 B
Python
25 lines
862 B
Python
import requests
|
|
|
|
def test_ollama_api(server_url, model_name, prompt):
|
|
url = f"{server_url}/api/generate"
|
|
headers = {"Content-Type": "application/json"}
|
|
data = {
|
|
"model": model_name,
|
|
"prompt": prompt
|
|
}
|
|
|
|
try:
|
|
response = requests.post(url, headers=headers, json=data)
|
|
response.raise_for_status() # HTTP 에러가 발생하면 예외 발생
|
|
result = response.json()
|
|
print("응답:", result["response"])
|
|
except requests.exceptions.RequestException as e:
|
|
print("API 요청 중 오류 발생:", e)
|
|
|
|
# 테스트 실행
|
|
server_url = "http://wrt08045.duckdns.org:11434" # 올라마 API 서버의 URL
|
|
model_name = "Llama3.2-Koren" # 사용할 모델 이름
|
|
prompt = "안녕하세요! 올라마 API를 테스트하고 있습니다."
|
|
|
|
test_ollama_api(server_url, model_name, prompt)
|