move prompt templates out of python bindings

This commit is contained in:
Jeffrey Morgan 2023-07-03 17:14:20 -04:00
parent 6093a88c1a
commit 9164981d72
16 changed files with 11 additions and 11 deletions

View file

@ -59,7 +59,7 @@ func NewCLI() *cobra.Command {
rootCmd := &cobra.Command{ rootCmd := &cobra.Command{
Use: "ollama", Use: "ollama",
Short: "Run any large language model on any machine.", Short: "Large language model runner",
CompletionOptions: cobra.CompletionOptions{ CompletionOptions: cobra.CompletionOptions{
DisableDefaultCmd: true, DisableDefaultCmd: true,
}, },
@ -88,14 +88,13 @@ func NewCLI() *cobra.Command {
}, },
} }
/*
runCmd := &cobra.Command{ runCmd := &cobra.Command{
Use: "run", Use: "run",
Short: "Run a model and submit prompts.", Short: "Run a model and submit prompts.",
RunE: func(cmd *cobra.Command. args []string) error { RunE: func(cmd *cobra.Command,args []string) error {
return nil
}, },
} }
*/
serveCmd := &cobra.Command{ serveCmd := &cobra.Command{
Use: "serve", Use: "serve",
@ -132,6 +131,7 @@ func NewCLI() *cobra.Command {
rootCmd.AddCommand( rootCmd.AddCommand(
modelsCmd, modelsCmd,
serveCmd, serveCmd,
runCmd,
) )
return rootCmd return rootCmd

View file

@ -23,5 +23,6 @@ target_include_directories(binding PRIVATE ${llama_cpp_SOURCE_DIR})
target_include_directories(binding PRIVATE ${llama_cpp_SOURCE_DIR}/examples) target_include_directories(binding PRIVATE ${llama_cpp_SOURCE_DIR}/examples)
target_link_libraries(binding llama ggml_static) target_link_libraries(binding llama ggml_static)
configure_file(${llama_cpp_SOURCE_DIR}/ggml-metal.metal ${CMAKE_CURRENT_BINARY_DIR}/ggml-metal.metal COPYONLY)
configure_file(${llama_cpp_BINARY_DIR}/libllama.a ${CMAKE_CURRENT_BINARY_DIR}/libllama.a COPYONLY) configure_file(${llama_cpp_BINARY_DIR}/libllama.a ${CMAKE_CURRENT_BINARY_DIR}/libllama.a COPYONLY)
configure_file(${llama_cpp_BINARY_DIR}/libggml_static.a ${CMAKE_CURRENT_BINARY_DIR}/libggml_static.a COPYONLY) configure_file(${llama_cpp_BINARY_DIR}/libggml_static.a ${CMAKE_CURRENT_BINARY_DIR}/libggml_static.a COPYONLY)

View file

@ -36,6 +36,8 @@ func Serve(ln net.Listener) error {
}) })
r.POST("/api/generate", func(c *gin.Context) { r.POST("/api/generate", func(c *gin.Context) {
// TODO: set prompt from template
var req api.GenerateRequest var req api.GenerateRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"message": err.Error()}) c.JSON(http.StatusBadRequest, gin.H{"message": err.Error()})
@ -64,13 +66,10 @@ func Serve(ln net.Listener) error {
return true return true
}) })
/* // embeds, err := l.Embeddings(text)
embeds, err := l.Embeddings(text) // if err != nil {
if err != nil { // fmt.Printf("Embeddings: error %s \n", err.Error())
fmt.Printf("Embeddings: error %s \n", err.Error()) // }
}
*/
}) })
log.Printf("Listening on %s", ln.Addr()) log.Printf("Listening on %s", ln.Addr())