highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-08-03 10:03:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1684 from dhiltgen/tag_integration_tests

Browse Source 
Guard integration tests with a tag
This commit is contained in:
Daniel Hiltgen
2023-12-22 16:43:41 -08:00
committed by GitHub
parent 10da41d677 697bea6939
commit cf50ef5b51
4 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/setup_integration_tests.sh
View File

@@ -2,6 +2,9 @@
# This script sets up integration tests which run the full stack to verify
# inference locally
#
# To run the relevant tests use
# go test -tags=integration ./server
set -e
set -o pipefail

2
server/llm_image_test.go
View File

@@ -1,3 +1,5 @@
//go:build integration
package server
import (

2
server/llm_test.go
View File

@@ -1,3 +1,5 @@
//go:build integration
package server
import (

5
server/llm_utils_test.go
View File

@@ -1,3 +1,5 @@
//go:build integration
package server
import (
@@ -38,7 +40,7 @@ func PrepareModelForPrompts(t *testing.T, modelName string, opts api.Options) (*
}
func OneShotPromptResponse(t *testing.T, ctx context.Context, req api.GenerateRequest, model *Model, runner llm.LLM) string {
prompt, err := model.Prompt(PromptVars{
prompt, err := model.PreResponsePrompt(PromptVars{
System: req.System,
Prompt: req.Prompt,
First: len(req.Context) == 0,
@@ -54,6 +56,7 @@ func OneShotPromptResponse(t *testing.T, ctx context.Context, req api.GenerateRe
success <- true
}
}
predictReq := llm.PredictOpts{
Prompt: prompt,
Format: req.Format,