Signed-off-by: Bryan Frimin <bryan@getprobo.com>
This commit is contained in:
Bryan Frimin
2026-03-16 00:28:47 +01:00
parent 7147cf189a
commit 6a77d42dd6
19 changed files with 21 additions and 37 deletions

View File

@@ -307,7 +307,6 @@ func toolCallResponse(toolCalls ...llm.ToolCall) *llm.ChatCompletionResponse {
} }
} }
func TestRun(t *testing.T) { func TestRun(t *testing.T) {
t.Parallel() t.Parallel()

View File

@@ -37,7 +37,7 @@ type NoOpHooks struct{}
var _ RunHooks = NoOpHooks{} var _ RunHooks = NoOpHooks{}
func (NoOpHooks) OnRunStart(context.Context, *Agent, []llm.Message) {} func (NoOpHooks) OnRunStart(context.Context, *Agent, []llm.Message) {}
func (NoOpHooks) OnRunEnd(context.Context, *Agent, *Result, error) {} func (NoOpHooks) OnRunEnd(context.Context, *Agent, *Result, error) {}
func (NoOpHooks) OnLLMStart(context.Context, *Agent, []llm.Message) {} func (NoOpHooks) OnLLMStart(context.Context, *Agent, []llm.Message) {}
func (NoOpHooks) OnLLMEnd(context.Context, *Agent, *llm.ChatCompletionResponse, error) {} func (NoOpHooks) OnLLMEnd(context.Context, *Agent, *llm.ChatCompletionResponse, error) {}
@@ -63,10 +63,10 @@ type NoOpAgentHooks struct{}
var _ AgentHooks = NoOpAgentHooks{} var _ AgentHooks = NoOpAgentHooks{}
func (NoOpAgentHooks) OnStart(context.Context, *Agent) {} func (NoOpAgentHooks) OnStart(context.Context, *Agent) {}
func (NoOpAgentHooks) OnEnd(context.Context, *Agent, string) {} func (NoOpAgentHooks) OnEnd(context.Context, *Agent, string) {}
func (NoOpAgentHooks) OnHandoff(context.Context, *Agent, *Agent) {} func (NoOpAgentHooks) OnHandoff(context.Context, *Agent, *Agent) {}
func (NoOpAgentHooks) OnToolStart(context.Context, *Agent, Tool) {} func (NoOpAgentHooks) OnToolStart(context.Context, *Agent, Tool) {}
func (NoOpAgentHooks) OnToolEnd(context.Context, *Agent, Tool, ToolResult) {} func (NoOpAgentHooks) OnToolEnd(context.Context, *Agent, Tool, ToolResult) {}
func (NoOpAgentHooks) OnLLMStart(context.Context, *Agent, []llm.Message) {} func (NoOpAgentHooks) OnLLMStart(context.Context, *Agent, []llm.Message) {}
func (NoOpAgentHooks) OnLLMEnd(context.Context, *Agent, *llm.ChatCompletionResponse, error) {} func (NoOpAgentHooks) OnLLMEnd(context.Context, *Agent, *llm.ChatCompletionResponse, error) {}

View File

@@ -25,4 +25,3 @@ type ModelSettings struct {
ToolChoice *llm.ToolChoice ToolChoice *llm.ToolChoice
ParallelToolCalls *bool ParallelToolCalls *bool
} }

View File

@@ -140,4 +140,3 @@ func (w *CacheStore) warmDomain(ctx context.Context, conn pg.Conn, domain *cored
return nil return nil
} }

View File

@@ -159,4 +159,3 @@ func (s *Selector) rebuildCacheEntry(ctx context.Context, conn pg.Conn, domain s
return nil return nil
} }

View File

@@ -565,4 +565,3 @@ WHERE
*sacs = controls *sacs = controls
return nil return nil
} }

View File

@@ -385,7 +385,6 @@ WHERE
return nil return nil
} }
func (a *Audits) LoadByControlID( func (a *Audits) LoadByControlID(
ctx context.Context, ctx context.Context,
conn pg.Conn, conn pg.Conn,

View File

@@ -396,4 +396,3 @@ WHERE %s
return nil return nil
} }

View File

@@ -98,4 +98,3 @@ WHERE
_, err := conn.Exec(ctx, q, args) _, err := conn.Exec(ctx, q, args)
return err return err
} }

View File

@@ -98,4 +98,3 @@ WHERE
_, err := conn.Exec(ctx, q, args) _, err := conn.Exec(ctx, q, args)
return err return err
} }

View File

@@ -345,4 +345,3 @@ LIMIT 1
return nil return nil
} }

View File

@@ -106,4 +106,3 @@ func NotEquals(key string, values ...string) Condition {
Values: values, Values: values,
} }
} }

View File

@@ -21,18 +21,18 @@ import (
type ( type (
ChatCompletionRequest struct { ChatCompletionRequest struct {
Model string Model string
Messages []Message Messages []Message
MaxTokens *int MaxTokens *int
Temperature *float64 Temperature *float64
TopP *float64 TopP *float64
FrequencyPenalty *float64 FrequencyPenalty *float64
PresencePenalty *float64 PresencePenalty *float64
StopSequences []string StopSequences []string
Tools []Tool Tools []Tool
ToolChoice *ToolChoice ToolChoice *ToolChoice
ParallelToolCalls *bool ParallelToolCalls *bool
ResponseFormat *ResponseFormat ResponseFormat *ResponseFormat
} }
ToolChoiceType string ToolChoiceType string
@@ -90,7 +90,7 @@ type (
} }
ChatCompletionStreamEvent struct { ChatCompletionStreamEvent struct {
Model string // present on first event if provider supports it Model string // present on first event if provider supports it
Delta MessageDelta Delta MessageDelta
Usage *Usage // present on final event if provider supports it Usage *Usage // present on final event if provider supports it
FinishReason *FinishReason // present on final event FinishReason *FinishReason // present on final event

View File

@@ -24,7 +24,6 @@ import (
"go.gearno.de/kit/pg" "go.gearno.de/kit/pg"
"go.probo.inc/probo/pkg/agents" "go.probo.inc/probo/pkg/agents"
"go.probo.inc/probo/pkg/certmanager" "go.probo.inc/probo/pkg/certmanager"
"go.probo.inc/probo/pkg/llm"
"go.probo.inc/probo/pkg/coredata" "go.probo.inc/probo/pkg/coredata"
"go.probo.inc/probo/pkg/crypto/cipher" "go.probo.inc/probo/pkg/crypto/cipher"
"go.probo.inc/probo/pkg/esign" "go.probo.inc/probo/pkg/esign"
@@ -33,6 +32,7 @@ import (
"go.probo.inc/probo/pkg/gid" "go.probo.inc/probo/pkg/gid"
"go.probo.inc/probo/pkg/html2pdf" "go.probo.inc/probo/pkg/html2pdf"
"go.probo.inc/probo/pkg/iam" "go.probo.inc/probo/pkg/iam"
"go.probo.inc/probo/pkg/llm"
"go.probo.inc/probo/pkg/mail" "go.probo.inc/probo/pkg/mail"
"go.probo.inc/probo/pkg/slack" "go.probo.inc/probo/pkg/slack"
) )

View File

@@ -69,4 +69,3 @@ func (sr *SafeRedirect) Redirect(w http.ResponseWriter, r *http.Request, redirec
safeURL := sr.GetSafeRedirectURL(redirectURL, fallbackURL) safeURL := sr.GetSafeRedirectURL(redirectURL, fallbackURL)
http.Redirect(w, r, safeURL, statusCode) http.Redirect(w, r, safeURL, statusCode)
} }

View File

@@ -254,4 +254,3 @@ func TestSafeRedirect_Redirect(t *testing.T) {
}) })
} }
} }

View File

@@ -58,4 +58,3 @@ func UnmarshalBigIntScalar(v any) (int64, error) {
return 0, fmt.Errorf("cannot unmarshal %T into BigInt", v) return 0, fmt.Errorf("cannot unmarshal %T into BigInt", v)
} }
} }

View File

@@ -77,4 +77,3 @@ func Max(max int) ValidatorFunc {
return nil return nil
} }
} }

View File

@@ -99,4 +99,3 @@ func UserAgent(component string) string {
return strings.Join(parts, " ") return strings.Join(parts, " ")
} }