mrrobot1024 commited on
Commit
3166780
·
1 Parent(s): 8acadd7

Removed the test files

Browse files
job_writing_agent/tools/test_llm.py DELETED
@@ -1,10 +0,0 @@
1
- from job_writing_agent.utils.llm_provider_factory import LLMFactory
2
- from langchain_cerebras import ChatCerebras
3
-
4
- llm_provider = LLMFactory()
5
-
6
- llm_cerebras = ChatCerebras(
7
- model="llama3.1-8b", # Direct name: "llama3.1-8b"
8
- temperature=0.3
9
- )
10
- print(llm_cerebras.invoke("Hey! Can you hear me?"))
 
 
 
 
 
 
 
 
 
 
 
job_writing_agent/tools/test_tavily.py DELETED
@@ -1,70 +0,0 @@
1
- from langchain_core.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate
2
- from langchain_core.prompt_values import PromptValue
3
- import dspy
4
- import os
5
- import mlflow
6
-
7
- mlflow.dspy.autolog(
8
- log_compiles=True,
9
- log_evals=True,
10
- log_traces_from_compile=True
11
- )
12
-
13
- mlflow.set_tracking_uri("http://127.0.0.1:5000/")
14
- mlflow.set_experiment("job description extract")
15
-
16
-
17
- OPENROUTER_API_KEY="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"
18
-
19
- dspy.configure(lm=dspy.LM(
20
- "openrouter/qwen/qwen3-4b:free",
21
- api_key=OPENROUTER_API_KEY,
22
- temperature=0.1
23
- ))
24
-
25
-
26
-
27
- class TavilySearchQueries(dspy.Signature):
28
- """Use the job description and company name
29
- to create search queries for the tavily search tool"""
30
- job_description_= dspy.InputField(desc="Job description of the role that candidate is applying for.")
31
- company_name = dspy.InputField(desc="Name of the company the candidate is applying for.")
32
- search_queries = dspy.OutputField(desc="Tavily Search Query")
33
- search_query_relevance = dspy.OutputField(desc="Relevance for each tavily search query that is generated")
34
-
35
-
36
- tavily_query_generator = dspy.ChainOfThought(TavilySearchQueries)
37
-
38
- job_description = """ Who are we?
39
-
40
- Our mission is to scale intelligence to serve humanity. We’re training and deploying frontier models for developers and enterprises who are building AI systems to power magical experiences like content generation, semantic search, RAG, and agents. We believe that our work is instrumental to the widespread adoption of AI.
41
-
42
- We obsess over what we build. Each one of us is responsible for contributing to increasing the capabilities of our models and the value they drive for our customers. We like to work hard and move fast to do what’s best for our customers.
43
-
44
- Cohere is a team of researchers, engineers, designers, and more, who are passionate about their craft. Each person is one of the best in the world at what they do. We believe that a diverse range of perspectives is a requirement for building great products.
45
-
46
- Join us on our mission and shape the future!
47
-
48
- About North:
49
-
50
- North is Cohere's cutting-edge AI workspace platform, designed to revolutionize the way enterprises utilize AI. It offers a secure and customizable environment, allowing companies to deploy AI while maintaining control over sensitive data. North integrates seamlessly with existing workflows, providing a trusted platform that connects AI agents with workplace tools and applications.
51
-
52
- As a Senior/Staff Backend Engineer, you will:
53
- Build and ship features for North, our AI workspace platform
54
- Develop autonomous agents that talk to sensitive enterprise data
55
- uns in low-resource environments, and has highly stringent deployment mechanisms
56
- As security and privacy are paramount, you will sometimes need to re-invent the
57
- wheel, and won’t be able to use the most popular libraries or tooling
58
- Collaborate with researchers to productionize state-of-the-art models and techniques
59
- You may be a good fit if:
60
- Have shipped (lots of) Python in production
61
- You have built and deployed extremely performant client-side or server-side RAG/agentic
62
- applications to millions of users You have strong coding abilities and are comfortable working across the stack. You’re able to read and understand, and even fix issues outside of the main code base
63
- You’ve worked in both large enterprises and startups
64
- You excel in fast-paced environments and can execute while priorities and objectives are a moving target
65
- If some of the above doesn’t line up perfectly with your experience, we still encourage you to apply!
66
- If you want to work really hard on a glorious mission with teammates that want the same thing, Cohere is the place for you."""
67
-
68
- response = tavily_query_generator(job_description_=job_description, company_name="Cohere")
69
-
70
- print(response)