OpenGradient-SDK/Makefile at main · OpenGradient/OpenGradient-SDK · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# OpenGradient SDK Makefile

# Default model for testing (override with: make chat MODEL=google/gemini-3-pro-preview)
MODEL ?= google/gemini-3-pro-preview

# ============================================================================
# Development
# ============================================================================

install:
	pip install -e .

build:
	python -m build

publish:
	@echo "Current version:" $$(grep 'version = ' pyproject.toml | cut -d'"' -f2)
	rm -rf dist/*
	python -m build
	twine upload dist/*

check:
	ruff format .
	mypy src
	mypy examples

docs:
	pdoc opengradient -o docs --template-dir ./templates --force

# ============================================================================
# Testing
# ============================================================================

test: utils_test client_test langchain_adapter_test opg_token_test

utils_test:
	pytest tests/utils_test.py -v

client_test:
	pytest tests/client_test.py -v

langchain_adapter_test:
	pytest tests/langchain_adapter_test.py -v

opg_token_test:
	pytest tests/opg_token_test.py -v

integrationtest:
	python integrationtest/agent/test_agent.py
	python integrationtest/workflow_models/test_workflow_models.py

examples:
	@for f in $$(find examples -name '*.py' | sort); do \
		echo "Running $$f..."; \
		python "$$f" || exit 1; \
	done
	@echo "All examples passed."

# ============================================================================
# CLI Examples (use MODEL=provider/model to change model)
# ============================================================================

infer:
	python -m opengradient.cli infer \
		-m "hJD2Ja3akZFt1A2LT-D_1oxOCz_OtuGYw4V9eE1m39M" \
		--input '{"open_high_low_close": [[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4],[1,2,3,4]]}'

completion:
	python -m opengradient.cli completion \
		--model $(MODEL) \
		--prompt "Hello, how are you?" \
		--max-tokens 50

chat:
	python -m opengradient.cli chat \
		--model $(MODEL) \
		--messages '[{"role":"user","content":"Tell me a fun fact"}]' \
		--max-tokens 350

chat-stream:
	python -m opengradient.cli chat \
		--model $(MODEL) \
		--messages '[{"role":"user","content":"Tell me a short story"}]' \
		--max-tokens 1250 \
		--stream

chat-tool:
	python -m opengradient.cli chat \
		--model $(MODEL) \
		--messages '[{"role":"user","content":"What is the weather in Tokyo?"}]' \
		--tools '[{"type":"function","function":{"name":"get_weather","description":"Get weather for a location","parameters":{"type":"object","properties":{"location":{"type":"string"},"unit":{"type":"string","enum":["celsius","fahrenheit"]}},"required":["location"]}}}]' \
		--max-tokens 100

chat-stream-tool:
	python -m opengradient.cli chat \
		--model $(MODEL) \
		--messages '[{"role":"user","content":"What is the weather in Tokyo?"}]' \
		--tools '[{"type":"function","function":{"name":"get_weather","description":"Get weather for a location","parameters":{"type":"object","properties":{"location":{"type":"string"},"unit":{"type":"string","enum":["celsius","fahrenheit"]}},"required":["location"]}}}]' \
		--max-tokens 100 \
		--stream

.PHONY: install build publish check docs test utils_test client_test langchain_adapter_test opg_token_test integrationtest examples \
	infer completion chat chat-stream chat-tool chat-stream-tool