common

type AppConfig struct {
	NumExecutionWorkers            int               `yaml:"numExecutionWorkers"`
	NumInferenceWorkers            int               `yaml:"numInferenceWorkers"`
	InferenceApiBaseUrls           []string          `yaml:"inferenceApiBaseUrls"`
	InferenceApiToken              string            `yaml:"inferenceApiToken"`
	ServerAddress                  string            `yaml:"serverAddress"`
	ServerPort                     string            `yaml:"serverPort"`
	ExpansionDepth                 int               `yaml:"expansionIntermediaryNodes"`
	PromptTemplates                map[string]string `yaml:"promptTemplates"`
	RegexTemplates                 map[string]string `yaml:"regexTemplates"`
	ExecutionContainers            map[string]string `yaml:"executionContainers"`
	ComputeEfficientMode           bool              `yaml:"useComputeEfficientMode"`
	ApplyRegexInferenceOnly        bool              `yaml:"applyRegexInferenceOnly"`
	EarlyStopOnTranslationSuccess  bool              `yaml:"earlyStop"`
	UseTranscoderTestFormat        bool              `yaml:"useTranscoderTestFormat"`
	VerifyIntermediateTranslations bool              `yaml:"verifyIntermediateTranslations"`
	StopOnDirectTranslation        bool              `yaml:"stopOnDirectTranslation"`
	UseIntermediatesMemoization    bool              `yaml:"useCrossPathIntermediatesMemoization"`
	UseInferenceCache              bool              `yaml:"useInferenceCache"`
	UseResponseCache               bool              `yaml:"useResponseCache"`
	UseExecutionCache              bool              `yaml:"useExecutionCache"`
	MaxGeneratedTokens             int               `yaml:"maxGeneratedTokens"`
	TopP                           float32           `yaml:"top-p"`
	TopK                           int               `yaml:"top-k"`
	Temperature                    float32           `yaml:"temperature"`
	Seed                           int               `yaml:"inferenceSeed"`
	DatabasePath                   string            `yaml:"cacheDatabasePath"`
	InferenceBackend               string            `yaml:"inferenceBackend"`
}

type BatchTranslationRequest struct {
	TranslationRequests []*TranslationRequest `protobuf:"bytes,1,rep,name=translation_requests,json=translationRequests,proto3" json:"translation_requests,omitempty"`
	Id                  string                `protobuf:"bytes,2,opt,name=id,proto3" json:"id,omitempty"`
	FileBaseName        string                `protobuf:"bytes,3,opt,name=file_base_name,json=fileBaseName,proto3" json:"file_base_name,omitempty"`
	FileSavePath        string                `protobuf:"bytes,4,opt,name=file_save_path,json=fileSavePath,proto3" json:"file_save_path,omitempty"`
	// contains filtered or unexported fields
}

type BatchTranslationResponse struct {
	TranslationResponses []*TranslationResponse `protobuf:"bytes,1,rep,name=translation_responses,json=translationResponses,proto3" json:"translation_responses,omitempty"`
	RequestId            string                 `protobuf:"bytes,2,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	ReturnedToDisk       bool                   `protobuf:"varint,3,opt,name=returnedToDisk,proto3" json:"returnedToDisk,omitempty"`
	// contains filtered or unexported fields
}

type BatchVerificationRequest struct {
	VerificationRequests []*VerificationRequest `protobuf:"bytes,1,rep,name=verification_requests,json=verificationRequests,proto3" json:"verification_requests,omitempty"`
	Id                   string                 `protobuf:"bytes,2,opt,name=id,proto3" json:"id,omitempty"`
	// contains filtered or unexported fields
}

type BatchVerificationResponse struct {
	VerificationRequests  *VerificationRequest    `protobuf:"bytes,1,opt,name=verification_requests,json=verificationRequests,proto3" json:"verification_requests,omitempty"`
	VerificationResponses []*VerificationResponse `protobuf:"bytes,2,rep,name=verification_responses,json=verificationResponses,proto3" json:"verification_responses,omitempty"`
	// contains filtered or unexported fields
}

type Counter struct {
	// contains filtered or unexported fields
}

type ExecutionType int

type ExecutionUnit struct {
	SourceCode         string
	Language           string
	StdinData          string
	ExecutionOutput    string
	Success            bool
	ExecutedCode       string
	OutputChannel      chan ExecutionUnit
	ExecutionType      ExecutionType
	WallTime           time.Duration
	UsedExecutionCache bool
}

type FuzzyTest struct {
	Input          string
	ExpectedOutput string
	ActualOutput   string
	Passed         bool
	ExecutedCode   string
	ExitCodeZero   bool
}

type FuzzyTestCase struct {
	StdinInput     string `protobuf:"bytes,1,opt,name=stdin_input,json=stdinInput,proto3" json:"stdin_input,omitempty"`
	ExpectedOutput string `protobuf:"bytes,2,opt,name=expected_output,json=expectedOutput,proto3" json:"expected_output,omitempty"`
	// contains filtered or unexported fields
}

type InferenceResult struct {
	Response string
	IsCached bool
	WallTime time.Duration
	Success  bool
}

type InferenceUnit struct {
	Prompt        string
	ModelName     string
	OutputChannel chan InferenceResult
	WallTime      time.Duration
}

type InfrastructureServiceClient interface {
	LaunchInferenceEndpoint(ctx context.Context, in *StartEndpointRequest, opts ...grpc.CallOption) (*LaunchResponse, error)
	StopInferenceEndpoint(ctx context.Context, in *StopEndpointRequest, opts ...grpc.CallOption) (*LaunchResponse, error)
}

type InfrastructureServiceServer interface {
	LaunchInferenceEndpoint(context.Context, *StartEndpointRequest) (*LaunchResponse, error)
	StopInferenceEndpoint(context.Context, *StopEndpointRequest) (*LaunchResponse, error)
	// contains filtered or unexported methods
}

type LaunchResponse struct {
	LaunchId int64 `protobuf:"varint,1,opt,name=launch_id,json=launchId,proto3" json:"launch_id,omitempty"`
	// contains filtered or unexported fields
}

type Path struct {
	Edges                 []*TranslationEdge
	FinalTarget           string
	PriorityPathWeight    int
	UsedMemoizedEdgeIndex []bool
}

type ResponseFuzzyTestCase struct {
	StdinInput     string `protobuf:"bytes,1,opt,name=stdin_input,json=stdinInput,proto3" json:"stdin_input,omitempty"`
	ExpectedOutput string `protobuf:"bytes,2,opt,name=expected_output,json=expectedOutput,proto3" json:"expected_output,omitempty"`
	ActualOutput   string `protobuf:"bytes,3,opt,name=actual_output,json=actualOutput,proto3" json:"actual_output,omitempty"`
	Passed         bool   `protobuf:"varint,4,opt,name=passed,proto3" json:"passed,omitempty"`
	ExecutedCode   string `protobuf:"bytes,5,opt,name=executed_code,json=executedCode,proto3" json:"executed_code,omitempty"`
	// contains filtered or unexported fields
}

type ResponseStatus int32

type ResponseTranslationEdge struct {
	PromptTemplate        string                   `protobuf:"bytes,1,opt,name=prompt_template,json=promptTemplate,proto3" json:"prompt_template,omitempty"`
	Prompt                string                   `protobuf:"bytes,2,opt,name=prompt,proto3" json:"prompt,omitempty"`
	TranslationId         string                   `protobuf:"bytes,3,opt,name=translation_id,json=translationId,proto3" json:"translation_id,omitempty"`
	InputLanguage         string                   `protobuf:"bytes,4,opt,name=input_language,json=inputLanguage,proto3" json:"input_language,omitempty"`
	TargetLanguage        string                   `protobuf:"bytes,5,opt,name=target_language,json=targetLanguage,proto3" json:"target_language,omitempty"`
	Level                 int32                    `protobuf:"varint,6,opt,name=level,proto3" json:"level,omitempty"`
	Success               bool                     `protobuf:"varint,7,opt,name=success,proto3" json:"success,omitempty"`
	InferenceOutput       string                   `protobuf:"bytes,8,opt,name=inference_output,json=inferenceOutput,proto3" json:"inference_output,omitempty"`
	ExecutionOutput       string                   `protobuf:"bytes,9,opt,name=execution_output,json=executionOutput,proto3" json:"execution_output,omitempty"`
	SourceCode            string                   `protobuf:"bytes,10,opt,name=source_code,json=sourceCode,proto3" json:"source_code,omitempty"`
	ExtractedSourceCode   string                   `protobuf:"bytes,11,opt,name=extracted_source_code,json=extractedSourceCode,proto3" json:"extracted_source_code,omitempty"`
	ParentEdgeId          int32                    `protobuf:"varint,12,opt,name=parent_edge_id,json=parentEdgeId,proto3" json:"parent_edge_id,omitempty"`
	Status                string                   `protobuf:"bytes,13,opt,name=status,proto3" json:"status,omitempty"`
	FuzzyTests            []*ResponseFuzzyTestCase `protobuf:"bytes,14,rep,name=fuzzy_tests,json=fuzzyTests,proto3" json:"fuzzy_tests,omitempty"`
	UnitTests             []*ResponseUnitTestCase  `protobuf:"bytes,15,rep,name=unit_tests,json=unitTests,proto3" json:"unit_tests,omitempty"`
	EdgeId                int32                    `protobuf:"varint,16,opt,name=edge_id,json=edgeId,proto3" json:"edge_id,omitempty"`
	WallTimeInference     int64                    `protobuf:"varint,17,opt,name=wallTimeInference,proto3" json:"wallTimeInference,omitempty"`
	WallTimeTestExecution int64                    `protobuf:"varint,18,opt,name=wallTimeTestExecution,proto3" json:"wallTimeTestExecution,omitempty"`
	UsedMemoization       bool                     `protobuf:"varint,19,opt,name=usedMemoization,proto3" json:"usedMemoization,omitempty"`
	UsedInferenceCache    bool                     `protobuf:"varint,20,opt,name=usedInferenceCache,proto3" json:"usedInferenceCache,omitempty"`
	// contains filtered or unexported fields
}

type ResponseTranslationPath struct {
	TranslationEdges  []*ResponseTranslationEdge `protobuf:"bytes,1,rep,name=translation_edges,json=translationEdges,proto3" json:"translation_edges,omitempty"`
	EdgeIndexMemoized []bool                     `protobuf:"varint,2,rep,packed,name=edge_index_memoized,json=edgeIndexMemoized,proto3" json:"edge_index_memoized,omitempty"`
	// contains filtered or unexported fields
}

type ResponseUnitTestCase struct {
	SourceCode   string `protobuf:"bytes,1,opt,name=source_code,json=sourceCode,proto3" json:"source_code,omitempty"`
	ActualOutput string `protobuf:"bytes,2,opt,name=actual_output,json=actualOutput,proto3" json:"actual_output,omitempty"`
	Passed       bool   `protobuf:"varint,3,opt,name=passed,proto3" json:"passed,omitempty"`
	ExecutedCode string `protobuf:"bytes,4,opt,name=executed_code,json=executedCode,proto3" json:"executed_code,omitempty"`
	// contains filtered or unexported fields
}

type StartEndpointRequest struct {
	ModelName string `protobuf:"bytes,1,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"`
	GpuId     string `protobuf:"bytes,2,opt,name=gpu_id,json=gpuId,proto3" json:"gpu_id,omitempty"`
	Port      string `protobuf:"bytes,3,opt,name=port,proto3" json:"port,omitempty"`
	Seed      int64  `protobuf:"varint,4,opt,name=seed,proto3" json:"seed,omitempty"`
	ApiToken  string `protobuf:"bytes,5,opt,name=api_token,json=apiToken,proto3" json:"api_token,omitempty"`
	LoraPath  string `protobuf:"bytes,6,opt,name=lora_path,json=loraPath,proto3" json:"lora_path,omitempty"`
	// contains filtered or unexported fields
}

type Status int

type StopEndpointRequest struct {
	LaunchId int64 `protobuf:"varint,1,opt,name=launch_id,json=launchId,proto3" json:"launch_id,omitempty"`
	// contains filtered or unexported fields
}

type TargetSignature struct {
	Language  string `protobuf:"bytes,1,opt,name=language,proto3" json:"language,omitempty"`
	Signature string `protobuf:"bytes,2,opt,name=signature,proto3" json:"signature,omitempty"`
	// contains filtered or unexported fields
}

type TestSuite struct {
	FuzzySuite    []*FuzzyTestCase `protobuf:"bytes,1,rep,name=fuzzy_suite,json=fuzzySuite,proto3" json:"fuzzy_suite,omitempty"`
	UnitTestSuite []*UnitTestCase  `protobuf:"bytes,2,rep,name=unit_test_suite,json=unitTestSuite,proto3" json:"unit_test_suite,omitempty"`
	// contains filtered or unexported fields
}

type TranslationEdge struct {
	Id                         int
	PromptTemplate             string
	Prompt                     string
	TranslationId              string
	InputLanguage              string
	TargetLanguage             string
	Level                      int
	Success                    bool
	InferenceOutput            string
	ExecutionOutput            string
	SourceCode                 string
	ExtractedSourceCode        string
	SuggestedTargetSignature   string
	RegexTemplate              string
	ModelName                  string
	ParentEdge                 *TranslationEdge
	FuzzyTests                 []FuzzyTest
	UnitTests                  []UnitTest
	WallClockInferenceTime     time.Duration
	WallClockTestExecutionTime time.Duration
	UsedMemoization            bool
	UsedInferenceCache         bool
	ExtraPromptData            string

	StatusMutex     *sync.Mutex // Mutex for thread safety
	ProcessingMutex *sync.Mutex
	// contains filtered or unexported fields
}

type TranslationPair struct {
	InputLanguage  string
	TargetLanguage string
}

type TranslationPaths struct {
	Paths []Path
}

type TranslationRequest struct {
	Id                 string             `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
	SeedLanguage       string             `protobuf:"bytes,2,opt,name=seed_language,json=seedLanguage,proto3" json:"seed_language,omitempty"`
	TargetLanguage     string             `protobuf:"bytes,3,opt,name=target_language,json=targetLanguage,proto3" json:"target_language,omitempty"`
	SeedCode           string             `protobuf:"bytes,4,opt,name=seed_code,json=seedCode,proto3" json:"seed_code,omitempty"`
	TestSuite          *TestSuite         `protobuf:"bytes,5,opt,name=test_suite,json=testSuite,proto3" json:"test_suite,omitempty"`
	UsedLanguages      []string           `protobuf:"bytes,6,rep,name=used_languages,json=usedLanguages,proto3" json:"used_languages,omitempty"`
	PromptTemplateName string             `protobuf:"bytes,7,opt,name=prompt_template_name,json=promptTemplateName,proto3" json:"prompt_template_name,omitempty"`
	TargetSignatures   []*TargetSignature `protobuf:"bytes,8,rep,name=target_signatures,json=targetSignatures,proto3" json:"target_signatures,omitempty"`
	RegexTemplateName  string             `protobuf:"bytes,9,opt,name=regex_template_name,json=regexTemplateName,proto3" json:"regex_template_name,omitempty"`
	ModelName          string             `protobuf:"bytes,10,opt,name=model_name,json=modelName,proto3" json:"model_name,omitempty"`
	ExtraPromptData    string             `protobuf:"bytes,11,opt,name=extra_prompt_data,json=extraPromptData,proto3" json:"extra_prompt_data,omitempty"`
	// contains filtered or unexported fields
}

type TranslationResponse struct {
	TranslationRequest *TranslationRequest        `protobuf:"bytes,1,opt,name=translation_request,json=translationRequest,proto3" json:"translation_request,omitempty"`
	Paths              []*ResponseTranslationPath `protobuf:"bytes,2,rep,name=paths,proto3" json:"paths,omitempty"`
	// contains filtered or unexported fields
}

type TranslationServiceClient interface {
	BatchTranslate(ctx context.Context, in *BatchTranslationRequest, opts ...grpc.CallOption) (*BatchTranslationResponse, error)
	BatchTranslateCAK(ctx context.Context, in *BatchTranslationRequest, opts ...grpc.CallOption) (*BatchTranslationResponse, error)
	BatchPanEtAlTranslate(ctx context.Context, in *BatchTranslationRequest, opts ...grpc.CallOption) (*BatchTranslationResponse, error)
	BatchRunVerification(ctx context.Context, in *BatchVerificationRequest, opts ...grpc.CallOption) (*BatchVerificationResponse, error)
}

type TranslationServiceServer interface {
	BatchTranslate(context.Context, *BatchTranslationRequest) (*BatchTranslationResponse, error)
	BatchTranslateCAK(context.Context, *BatchTranslationRequest) (*BatchTranslationResponse, error)
	BatchPanEtAlTranslate(context.Context, *BatchTranslationRequest) (*BatchTranslationResponse, error)
	BatchRunVerification(context.Context, *BatchVerificationRequest) (*BatchVerificationResponse, error)
	// contains filtered or unexported methods
}

type TreeBranch struct {
	TranslationEdges   []TranslationEdge
	PriorityScheduling bool
}

type UnimplementedInfrastructureServiceServer struct {
}

type UnimplementedTranslationServiceServer struct {
}

type UnitTest struct {
	SourceCode   string
	ActualOutput string
	ExecutedCode string
	Passed       bool
	Imports      string
	ExitCodeZero bool
}

type UnitTestCase struct {
	Language string `protobuf:"bytes,1,opt,name=language,proto3" json:"language,omitempty"`
	TestCase string `protobuf:"bytes,2,opt,name=test_case,json=testCase,proto3" json:"test_case,omitempty"`
	Imports  string `protobuf:"bytes,3,opt,name=imports,proto3" json:"imports,omitempty"`
	// contains filtered or unexported fields
}

type UnsafeInfrastructureServiceServer interface {
	// contains filtered or unexported methods
}

type UnsafeTranslationServiceServer interface {
	// contains filtered or unexported methods
}

type VerificationRequest struct {
	Id              string     `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
	TestSuite       *TestSuite `protobuf:"bytes,2,opt,name=test_suite,json=testSuite,proto3" json:"test_suite,omitempty"`
	InferenceOutput string     `protobuf:"bytes,3,opt,name=inferenceOutput,proto3" json:"inferenceOutput,omitempty"`
	TargetLanguage  string     `protobuf:"bytes,4,opt,name=targetLanguage,proto3" json:"targetLanguage,omitempty"`
	SourceLanguage  string     `protobuf:"bytes,5,opt,name=sourceLanguage,proto3" json:"sourceLanguage,omitempty"`
	// contains filtered or unexported fields
}

type VerificationResponse struct {
	VerificationRequest *VerificationRequest     `protobuf:"bytes,1,opt,name=verification_request,json=verificationRequest,proto3" json:"verification_request,omitempty"`
	FuzzyTests          []*ResponseFuzzyTestCase `protobuf:"bytes,2,rep,name=fuzzy_tests,json=fuzzyTests,proto3" json:"fuzzy_tests,omitempty"`
	UnitTests           []*ResponseUnitTestCase  `protobuf:"bytes,3,rep,name=unit_tests,json=unitTests,proto3" json:"unit_tests,omitempty"`
	Status              string                   `protobuf:"bytes,6,opt,name=status,proto3" json:"status,omitempty"`
	// contains filtered or unexported fields
}